easy_ViTPose/vit_utils/nms/gpu_nms.hpp

void _nms(int* keep_out, int* num_out, const float* boxes_host, int boxes_num,
          int boxes_dim, float nms_overlap_thresh, int device_id);