Mmdetection踩坑教程

Mmdetection踩坑指南

  • 菜鸟初学mmdetection没几天,遇到一堆坑,和大家交流下
    • 1.Source not found
    • 2.libSM.so.6
    • 3.import name 'deform_conv_cuda'
    • 向paperspace上传文件
    • dtype torch.uint8现在是 不推荐使用,请改用dtype torch.bool(这是中文翻译hhhh)
    • mmdetection训练模型
    • 测试模型
    • 评估mAP
      • 显存不足 GPU out of memory

菜鸟初学mmdetection没几天,遇到一堆坑,和大家交流下

我用的是paperspace,可以不会临时更改环境,稳定一点,但是免费GPU一般
配置环境请移步官方指导或者 知乎用户指南.

1.Source not found

exec bash来从dash切换到bash,之后就能正常使用source

2.libSM.so.6

sudo apt-get install libsm6 libxrender1 libfontconfig1

3.import name ‘deform_conv_cuda’

环境没配好 pytorch0.4.1不能用,需要使用1.1.0然后进行./compile.sh(各种指南都坑了一遍,自己瞎搞忽然发现0.4.1的步骤+1.1.0的步骤就好了,emmmmmp),然后python setup.py develop即可

向paperspace上传文件

解决链接.

dtype torch.uint8现在是 不推荐使用,请改用dtype torch.bool(这是中文翻译hhhh)

在train.py文件开头加上
import warnings
warnings.filterwarnings(“ignore”)

mmdetection训练模型

去博客设置页面,选择一款你喜欢的代码片高亮样式,下面展示同样高亮的 代码片.

#从开始训练模型
python tools/train.py configs/xxx.py
#从断点权重文件继续训练
python tools/train.py configs/xxx.py --resume_from="work_dirs/yy.pth"

测试模型

python tools/test.py configs/xxx.py  work_dirs/yy.pth --out ./result/zzz.pkl

评估mAP

python tools/voc_eval.py result/zzz.pkl configs/xxx.py

显存不足 GPU out of memory

nvidia-smi#查看GPU当前状态
ps aux
Kill -9 [PID]

你可能感兴趣的:(mmdetection,mmdetection踩坑教程)