faster-rcnn.pytorch-pytorch-1.0更换数据集后遇到KeyError: ‘tumour‘

原数据集为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')

成功解决问题