Check out my code for fine tuning the keypointrcnn_resnet50_fpn model for detecting 2 keypoints.

  
      github.com
  
  
    sandmantan90/Keypoint-RCNN/blob/master/keypoint_CNN1.1.ipynb 137
{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "name": "keypoint_CNN.ipynb",
      "provenance": [],
      "collapsed_sections": []
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    },
    "accelerator": "GPU",
    "widgets": {
      "application/vnd.jupyter.widget-state+json": {
        "53320d6fe6eb4cedad0a6d3ff9529650": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "state": {


  This file has been truncated. show original

  
  
    
    
  
  


It works!