现在网上大多数自己手写的NMS都会导致GPU利用率较低，进而导致训练的时间较长，这是因为会经常调用单个box和其他box之间的IOU计算，导致内存和显存之间的IO交互较多。 def nms(boxes, scores, iou_threshold): keep = [] idxs = scores.argsort() ious = circle_iou(box
#[derive(Clone, Copy)] enum Args<'a> { Idx(usize), IdxList(&'a [usize]), } fn get_data<'a, T>(arr: &'a [T], idxs: Args<'a>) -> Vec<&'a T> { match idxs { Args::Idx(id
