M2Det 尝试
arXiv:?https://arxiv.org/abs/1811.04533
github: https://github.com/qijiezhao/M2Det
作者團(tuán)隊(duì):北京大學(xué)和阿里巴巴
注:AAAI 2019
一、創(chuàng)建環(huán)境
conda create -n M2Det python=3.7命名一個(gè)環(huán)境M2Det,python版本3.7
conda activate M2Det啟動(dòng)環(huán)境
conda install pytorch==0.4.1 torchvision -c pytorch pip install opencv-python tqdm安裝需要的模塊(國(guó)內(nèi)的鏡像貌似都不能用了,求ubuntu科學(xué)上網(wǎng)的方法,下載真是要人命)
pytorch模塊一直下載不下來(lái),用為windows上網(wǎng)下載下來(lái)包,然后拷貝過(guò)來(lái)安裝
安裝包忽然不能重命名,要不然報(bào)錯(cuò)
conda install 安裝包名稱(chēng) git clone https://github.com/qijiezhao/M2Det.git下載項(xiàng)目并打開(kāi)
sh make.shmake一下
二、Demo
(本部分參考:https://blog.csdn.net/c20081052/article/details/88803408)
下載訓(xùn)練好的模型,baidu cloud,google drive
運(yùn)行
python demo.py -c=configs/m2det512_vgg.py -m=weights/m2det512_vgg.pth --show報(bào)錯(cuò) ,安裝了一個(gè)DEBUG工具(大概)
conda install --debug mkl?再運(yùn)行,報(bào)錯(cuò)
pip install addictconda下找不到addict模塊,pip安裝
conda install torchvision opencv-python缺啥裝啥吧 ,懶得一個(gè)一個(gè)打
又發(fā)現(xiàn)一個(gè)錯(cuò)誤,找不到權(quán)重文件,折騰了半天,最后把下載好的權(quán)重文件重命名了一遍解決了。。。
?結(jié)果
視頻檢測(cè)
?如圖,有誤檢,重檢
三、訓(xùn)練自己的樣本
總結(jié)
- 上一篇: Eclipse 优化
- 下一篇: 【AI安全】对抗样本之FGSM的代码实现