将DIOU,CIOU,SIOU用到YOLOX中

概述:本文使用pytorch实现将DIOU,CIOU和SIOU接入YOLOX中,并验证了可以正常训练,关于这3种IOU的介绍这里就不赘述了,感兴趣的朋友可自行阅读原论文。

话不多说,直接进入正题。在YOLOX/yolox/model/losses.py中添加如下代码:将DIOU,CIOU,SIOU用到YOLOX中_第1张图片

将DIOU,CIOU,SIOU用到YOLOX中_第2张图片将DIOU,CIOU,SIOU用到YOLOX中_第3张图片 

将DIOU,CIOU,SIOU用到YOLOX中_第4张图片 

 结语:目前我正在实验IOU,CIOU和SIOU的效果,希望SIOU真有效果!(球球了)!对结果感兴趣的朋友可以留言交流!

 

你可能感兴趣的:(人工智能项目实战,深度学习,计算机视觉,神经网络)