文件配置训练的过程可以看
AssertionError: The `num_classes` (3) in Shared2FCBBoxHead of MMDataParallel does not matches the length of `CLASSES` 80) in CocoDataset
解决方案 :
_base_ = [
'./faster_rcnn_r50_fpn_1x_coco.py'
]
print(_base_)
model = dict(
roi_head=dict(
bbox_head = dict(num_classes=4)
)
)
dataset_type = 'COCODataset'
classes = ('',' id','','') #你自己数据集的类别
#你自己数据集的位置
data = dict(
train = dict(
img_prefix = '../../dataset/train/',
classes = classes,
ann_file= '../../dataset/train/_annotations.coco.json'),
val = dict(
img_prefix = '../../dataset/val/',
classes = classes,
ann_file= '../../dataset/val/_annotations.coco.json'),
)
开始训练
python tools/train.py custom_faster_rcnn/faster_rcnn.py --work-dir custom_faster_rcnn/work_dirs/faster_rcnn
然后运行一下中断
打开work_dirs文件
进行调参