code: def __init__(self, use_heatmap_max_keypoint):
      self.use_heatmap_max_keypoint = use_heatmap_max_keypoint
  
imports: import math
  from typing import Dict
  import torch
  import torch.nn.functional
  from detectron2.layers import ShapeSpec, cat
  from detectron2.layers.roi_align_rotated import ROIAlignRotated
  from detectron2.modeling import poolers
  from detectron2.modeling.proposal_generator import rpn
  from detectron2.modeling.roi_heads.mask_head import mask_rcnn_inference
  from detectron2.structures import Boxes, ImageList, Instances, Keypoints, RotatedBoxes
  from .shared import alias, to_device
label_desc: Use image node_id 1 to create a new Caffe2KeypointRCNNInference object  with example: obj = Caffe2KeypointRCNNInference(use_heatmap_max_keypoint)