Spaces:

shvardhan
/

object_detection

Runtime error

shvardhan commited on Mar 4, 2024

Commit

8f53050

•

1 Parent(s): 6c33fe7

update config file

Files changed (1) hide show

configs/_base_/faster-rcnn_r50_fpn_1x_coco.py CHANGED Viewed

@@ -44,7 +44,7 @@ model = dict(
             in_channels=256,
             fc_out_channels=1024,
             roi_feat_size=7,
-            num_classes=80,
             bbox_coder=dict(
                 type='DeltaXYWHBBoxCoder',
                 target_means=[0., 0., 0., 0.],

             in_channels=256,
             fc_out_channels=1024,
             roi_feat_size=7,
+            num_classes=1,
             bbox_coder=dict(
                 type='DeltaXYWHBBoxCoder',
                 target_means=[0., 0., 0., 0.],