-
Notifications
You must be signed in to change notification settings - Fork 5.5k
keypoints detection fails #120
Comments
Hi @olgaliak, to get keypoint results you can use one of the keypoint models from the Model Zoo. For example, you can run e2e Keypoint R-CNN with the default ResNet-101 config using:
Please note also that you should not change values in the |
Thanks a lot for super quick response! |
Hi ================== |
The same error with @srikanth-kilaru |
I'm running infer_simple.py with MaskRCNN and would like to get keypoints detection results.
Here are config changes I made:
__C.MODEL.KEYPOINTS_ON = True
__C.KRCNN.ROI_KEYPOINTS_HEAD = b'keypoint_rcnn_heads.add_roi_pose_head_v1convX'
Here is cmd (running in docker container):
python2 tools/infer_simple.py --cfg configs/12_2017_baselines/e2e_mask_rcnn_R-101-FPN_2x.yaml --output-dir /tmp/detectron-visualizations --image-ext jpeg --wts https://s3-us-west-2.amazonaws.com/detectron/35861858/12_2017_baselines/e2e_mask_rcnn_R-101-FPN_2x.yaml.02_32_51.SgT4y1cO/output/train/coco_2014_train:coco_2014_valminusminival/generalized_rcnn/model_final.pkl demo2
Actual results
here is output:
System information
The text was updated successfully, but these errors were encountered: