1. mmseg中图片格式高宽要一致
2. mmdet中的mask rcnn如何出现mmsegm无法计算,可注释掉base中验证的dataset的计算mmseg。
3. 源码改动:
mmdet:class_names
mmsegm: 背景*255
mmclass: 分类数
4. 跑程序,跑一个epoch就要看结果
5. linux改文件权限: chmod -R 777 mmdet
6. CUDA_VISIBLE_DEVICES=0,1,2,3 python XXX.py
指定GPU运行
7. 当出现KeyError: ‘***** is not in the dataset registry‘时,
可能由于dataset里的库是其他人安装的,可以执行 pip install -v -e .
重新安装一下。这时,mmcls/dataset中的库变为自己的dataset.
9. 对于推理时,a=result[0],对于YOLO算法的检测框,得到物体的数量可以用a.shape[0];
对于mask_rcnn算法的检测框,需要多加一步得到bbox检测框,a[0],然后a[0].shape[0].
10. list 用append依次加入
11. 模型推理时,result[0] 是bbox框,
result[1][0][i]是mask的掩膜,数据为bool值,true或flase
12. 模型推理时,用demo/image_demo时,需要最后一个参数--out-file=demo.jpg
mmlab代码调试(学习记录)
于 2023-11-08 13:51:51 首次发布