mmdet 修改类别和数据集

下面的内容以faster_rcnn_r50_fpn_1x_coco.py为例

修改数据集

1. 由于我用到的数据集格式是voc，但默认的是coco格式，现在配置文件中修改数据集。
   

2. 在configs/_base_/datasets/voc0712.py文件中，修改数据集路径和voc07或者voc12。我这里只有voc2007的数据集。
   
   

修改类别

1. config 文件中，修改 num_classes=数据集中的类别数。
2. 在mmdet/datasets/voc.py和mmdet/core/evaluation/class_names中修改对应的类别。

遇到的错误

unexpected key in source state_dict: fc.weight, fc.bias

Segmentation fault (core dumped)

可能是cuda版本的问题，检查了一下服务器的版本是CUDA Version: 11.4，换成CUDA Version: 11.1就可以正常运行了。