1.在训练框架下的faster-rcnn时,报错:
ValueError: need at least one array to concatenate
先要确保自己的配置文件已经修改好了,数据集路径是否正确,这些都没问题的话,只需要找到你的虚拟环境,在Lib/site-packages/mmdet下修改core/evaluation/class_names.py文件:
def coco_classes():
return ['airport', ] #单类别,需要结尾加逗号
然后修改虚拟环境Lib/site-packages/mmdet底下的datasets/coco.py文件:
CLASSES = ('person', ) #单类别,需要结尾加逗号
2.训练mask-rcnn时报错:
AssertionError: The `num_classes` (1) in Shared2FCBBoxHead of MMDataParallel does not matches the length of `CLASSES` 80) in CocoDataset
解决方法同上。