yancx

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

摘要： 一种CIOU、GIOU和DIOU的实现，可以计算不同尺寸的tensor数据 有一些方法来自于torchvision的官方实现 def box_ciou(boxes1, boxes2, GIOU=False, CIOU=True, DIOU=False): """ Return complete IO 阅读全文