下面的内容以faster_rcnn_r50_fpn_1x_coco.py
为例
修改数据集
-
由于我用到的数据集格式是voc,但默认的是coco格式,现在配置文件中修改数据集。
-
在
configs/_base_/datasets/voc0712.py
文件中,修改数据集路径和voc07或者voc12。我这里只有voc2007的数据集。
修改类别
- config 文件中,修改
num_classes=数据集中的类别数
。 - 在
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
就可以正常运行了。