原数据集为VOCdevkit2007,在更换乳腺癌数据集后,训练模型的时候遇到了KeyError: 'tumour'的问题。这是类名不匹配的问题导致的。在lib\datasets\pascal_voc.py文件中把
self._classes = ('__background__', # always index 0
'aeroplane', 'bicycle', 'bird', 'boat',
'bottle', 'bus', 'car', 'cat', 'chair',
'cow', 'diningtable', 'dog', 'horse',
'motorbike', 'person', 'pottedplant',
'sheep', 'sofa', 'train', 'tvmonitor')
更换为
self._classes = ('__background__',
'tumour')
成功解决问题