MM-Detection环境配置
前置环境配置
pip install -U openmim -i https://pypi.tuna.tsinghua.edu.cn/simple
mim install mmengine -i https://pypi.tuna.tsinghua.edu.cn/simple
mim install "mmcv>=2.0.0" -i https://pypi.tuna.tsinghua.edu.cn/simple
下载mmdetection源码
https://github.com/open-mmlab/mmdetection
mim install -v -e . -i https://pypi.tuna.tsinghua.edu.cn/simple
mim download mmdet --config yolox_s_8x8_300e_coco --dest .
验证安装
步骤 1. 我们需要下载配置文件和模型权重文件。
mim download mmdet --config rtmdet_tiny_8xb32-300e_coco --dest .
步骤 2. 推理验证。
python demo/image_demo.py demo/demo.jpg rtmdet_tiny_8xb32-300e_coco.py --weights rtmdet_tiny_8xb32-300e_coco_20220902_112414-78e30dcc.pth --device cpu
补充:
pip install numpy==1.20.0 -i https://pypi.tuna.tsinghua.edu.cn/simple