随笔分类 -  深度学习-代码实现

摘要:现在网上大多数自己手写的NMS都会导致GPU利用率较低,进而导致训练的时间较长,这是因为会经常调用单个box和其他box之间的IOU计算,导致内存和显存之间的IO交互较多。 def nms(boxes, scores, iou_threshold): keep = [] idxs = scores. 阅读全文
posted @ 2022-07-31 17:47 Ayu000 阅读(434) 评论(0) 推荐(0)