Skip to content

Commit c25b53d

Browse files
ppwwyyxxfacebook-github-bot
authored andcommitted
fix num_classes in keypoint rcnn test
Reviewed By: alexander-kirillov Differential Revision: D26472684 fbshipit-source-id: c7ac903a6c29f4650c63e8f448950582981bb9a8
1 parent 1f522c1 commit c25b53d

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

configs/quick_schedules/keypoint_rcnn_R_50_FPN_instant_test.yaml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,8 @@ _BASE_: "../Base-RCNN-FPN.yaml"
22
MODEL:
33
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
44
KEYPOINT_ON: True
5+
ROI_HEADS:
6+
NUM_CLASSES: 1
57
DATASETS:
68
TRAIN: ("keypoints_coco_2017_val_100",)
79
TEST: ("keypoints_coco_2017_val_100",)

0 commit comments

Comments
 (0)