Skip to content
Snippets Groups Projects
Unverified Commit 72a9146e authored by mrbulb's avatar mrbulb Committed by GitHub
Browse files

[fix] fix num_classes bug in faster rcnn config (#4088)

`num_classes` in model should be the same as the number of dataset class
parent 7ad44cd6
No related branches found
No related tags found
No related merge requests found
_base_ = './faster_rcnn_r50_fpn_1x_coco.py'
model = dict(roi_head=dict(bbox_head=dict(num_classes=3)))
classes = ('person', 'bicycle', 'car')
data = dict(
train=dict(classes=classes),
......
_base_ = './faster_rcnn_r50_fpn_1x_coco.py'
model = dict(roi_head=dict(bbox_head=dict(num_classes=1)))
classes = ('person', )
data = dict(
train=dict(classes=classes),
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment