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