How to find the keypoints for using "draw_and_connect_keypoints(keypoints)"

[facebookresearch]

How to find the keypoints for using "draw_and_connect_keypoints(keypoints)"

[ppwwyyxx]

You need a different model.

cfg = get_cfg()
cfg.merge_from_file("./detectron2_repo/configs/COCO-Keypoints/keypoint_rcnn_R_50_FPN_3x.yaml")
cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.7  # set threshold for this model
cfg.MODEL.WEIGHTS = "detectron2://COCO-Keypoints/keypoint_rcnn_R_50_FPN_3x/137849621/model_final_a6e10b.pkl"
predictor = DefaultPredictor(cfg)
outputs = predictor(im)
v = Visualizer(im[:,:,::-1], MetadataCatalog.get(cfg.DATASETS.TRAIN[0]), scale=1.5)
v = v.draw_instance_predictions(outputs["instances"].to("cpu"))
cv2_imshow(v.get_image()[:, :, ::-1])

(this is now included in colab)

[ppwwyyxx]

You can use any combination of config + model in https://github.com/facebookresearch/detectron2/blob/master/MODEL_ZOO.md#coco-person-keypoint-detection-baselines-with-keypoint-r-cnn

[AntonBaumannDE]

Great!! I couldnt find the right path for the weights, so this is perfect!! Thanks a lot! Made my day