首页 > TAG信息列表 > idxs

NMS(解决大多数手写NMS运算效率较低的问题)

现在网上大多数自己手写的NMS都会导致GPU利用率较低，进而导致训练的时间较长，这是因为会经常调用单个box和其他box之间的IOU计算，导致内存和显存之间的IO交互较多。   def nms(boxes, scores, iou_threshold): keep = [] idxs = scores.argsort() ious = circle_iou(box

rust 变参

#[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