在安装mmdetection之后,我们可能会按照官方文档来验证是否安装成功。而验证的第一步是下载预训练模型,运行命令为:
mim download mmdet --config yolov3_mobilenetv2_320_300e_coco --dest .
运行完之后,在当前目录下会下载到两个文件,即yolov3_mobilenetv2_320_300e_coco.py 和yolov3_mobilenetv2_320_300e_coco_20210719_215349-d18dff72.pth。
如果运行文件时,提示search.py的392行抛出如下异常:
KeyError: 'Cascade Mask R-CNN'
替换399行即可:
if collection_name and collection_name in name2collection.keys():