方案一:有人说 在自己定义的 conifg文件中增加
metainfo = { 'classes': ('class1','class2', 'class2',), 'palette': [ (220, 20, 60), (221, 11, 22),(221, 11, 42), ] }
方案二:修改mmdet/evaluation/metrics 文件的内容,mmdet 是 python/site-package下的 mmdet
1 修改 mmdet/evaluation/functional/class_names.py
def coco_classes() -> list: """Class names of COCO.""" return [ 'clas1','class2','class3', ]
2 修改 mmdet/datasets/coco.py
METAINFO = { 'classes':('class1','class2','class3',), # palette is a list of color tuples, which is used for visualization. 'palette': [ (220, 20, 60), (221, 11, 22),(221, 11, 42),] }
方案三:我试了一下,还是没有用,感觉这是MMdetection的一个BUG,然后分别打印出来 报错中的 label self.cat_ids,
1 在完成方案一和方案二之后 如果label超出了你设置的类数量范围,可以检查config文件中关于model原始的config(一般在_base_文件夹下的model子文件夹下)的 num_classes=3;
2 如果是没有问题的;那肯定就是 self.cat_ids的维度小于你设置的类的数量,那可以选择简单粗暴一些,直接在mmdet/evaluation/metrics/coco_metric.py 中的 def compute_metrics(self, results: list) -> Dict[str, float]:部分,直接修改 self.cat_ids = [0, 1] ;问题肯定就被就解决了