yolov5 WIOU
时间: 2023-12-03 13:40:58 浏览: 97
yolov5
Yolov5 WIOU是一种基于IoU的目标检测算法,它是在Yolov5的基础上增加了Wasserstein距离的计算,可以更好地处理目标检测中的类别不平衡问题。WIOU的计算方式是将IoU和Wasserstein距离相结合,其中Wasserstein距离是一种用于度量两个概率分布之间距离的方法。在Yolov5中,WIOU的计算方式是在bbox_iou函数中增加了Wasserstein距离的计算,具体实现可以参考Yolov5的源代码。
阅读全文