【VisDrone|目标检测】保姆级教程：Cascade-RCNN模型训练VisDrone数据集（含代码修改+训练启动）（四）

一、修改两处代码

第一处： mmdet/evalution/class_names.py 代码下的 def coco_classes() 的 return 内容改为自己数据集的类别；

第二处：mmdet/datasets/coco.py 代码下的 class CocoDataset(CustomDataset) 的 CLASSES 改为自己数据集的类别；

注意：修改两处后，一定要在根目录下，输入命令：

python setup.py install build

重新编译代码，要不然类别会没有载入，还是原coco类别，训练异常