第二步;在utils/loss.py中,找到ComputeLoss类中的__call__()函数,把Regression loss中计算iou的代码,换成下面这句: iou=bbox_iou(pbox.T,tbox[i],x1y1x2y2=False,CIoU=False,EIoU=True)# iou(prediction, target) 2.alpha IoU更换方式 第一步;直接将utils/metrics.py文件中bbox_iou()替换,随后将b...
以YOLOv5为基础网络,通过在基础网络结构中替换损失函数和增加注意力机制模块来改善目标检测问题.用SIoU Loss损失函数替换CIoU Loss损失函数,对距离损失进行重新定义,... 程国建,李嘉欣,贾强 - 《石油工业技术监督》 被引量: 0发表: 2024年 一种基于改进型YOLOV5算法的荔枝晚秋梢检测方法及处理终端 本发明公开了一...