准备工作:
conda create -n openmmlab python=3.7 pytorch==1.6.0 cudatoolkit=10.1 torchvision -c pytorch -y
conda activate openmmlab
pip install openmim
mim install mmcv-full
git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -r requirements/build.txt
pip install -v -e .
选图片(bus.jpg):
选预训练模型:
点击红框即可下载
这里我选用hrnet,cascade_rcnn_hrnetv2p_w40_20e_coco_20200512_161112-75e47b04.pth
新建文件夹,checkpoints,把cascade_rcnn_hrn