code: def get_close_kpt_num(g, d, dis_thresh=1):
      """Calculate number of close keypoints"""
      xg = g[0::3]
      yg = g[1::3]
      close_keypoints_num = np.zeros((d.shape[0]))
      for n_d in range(0, d.shape[0]):
          xd = d[n_d, 0::3]
          yd = d[n_d, 1::3]
          dx = xd - xg
          dy = yd - yg
          dis = np.sqrt(dx**2 + dy**2)
          close_keypoints_num[n_d] = (dis <= dis_thresh).sum()
      return close_keypoints_num
  
file_path: gluon-cv/gluoncv/torch/nn/nms.py
imports: import numpy
  import torch
  from torchvision.ops import boxes
  from torchvision.ops import nms
label_desc: Use image node_id 4 for calling a global function with example usage: get_close_kpt_num(g, d, dis_thresh) and returns: close_keypoints_num