mmdection学习笔记1: mmdection环境配置,运行demo
mmdection环境配置
pip install -U openmim
mim install mmengine
mim install "mmcv>=2.0.0"
git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -v -e .
过程安装不报错,则环境搭建完成。我们测试一下:
Demo测试
图形识别demo:
Example:
Save visualizations and predictions results::
python demo/image_demo.py demo/demo.jpg rtmdet-s
python demo/image_demo.py demo/demo.jpg \
configs/rtmdet/rtmdet_s_8xb32-300e_coco.py \
--weights rtmdet_s_8xb32-300e_coco_20220905_161602-387a891e.pth \
--out-dir /home/jack/mmaction2/mmdetection/outputs/vis/
图形识别准确性比较高。
ERROR处理:
问题报错:FileNotFoundError: checkpoints/rtmdet_l_8xb32-300e_coco_20220719_112030-5a0be7c4.pth can not be found.
解决方案:https://github.com/open-mmlab/mmdetection/tree/main/configs/rtmdet ,pth缺失,下载指定的checkpoints文件(model)即可。同理若提示config/xxxx.py没找到一样解决,到本链接地址下载config就可以了。
视频识别demo:
视频识别
python demo/video_gpuaccel_demo.py demo/demo.mp4 \
configs/rtmdet/rtmdet_l_8xb32-300e_coco.py \
checkpoints/rtmdet_l_8xb32-300e_coco_20220719_112030-5a0be7c4.pth \
--nvdecode --out result.mp4
视频识别有点小问题,画的框不是很准。还需要进一步测试一下。
ERROR处理:
1.FileNotFoundError: checkpoints/rtmdet_l_8xb32-300e_coco_20220719_112030-5a0be7c4.pth can not be found.
—去官网下载模型
2.nvdecode报错。此为加速器,删除即可。不影响程序运行。
总结
mmdection环境配置相对比较容易。CMD界面运行时,只要关注py引导文件、输入文件路径、config路径、checkpoints路径、输出路径,程序一般就可以正常运行了、