search

yangxue0827 / RotationDetection

This is a tensorflow-based rotation detection benchmark, also called AlphaRotate.
https://rotationdetection.readthedocs.io/
Apache License 2.0
1.07k stars 180 forks source link

KFIou疑问 #112

Open LUO77123 opened 2 years ago

LUO77123 commented 2 years ago

大佬您好，想请问一下，我将KFIou移植到yolov5上去，但是训练几十张图就开始变成nan,查了一下，也不太清楚里面具体原因，大佬可以指导一下吗？谢谢您！（单步调试发现return的loss大于1的情况比较多，不管KFIou值在0-1/3 或者（乘以3，值在）0-1都一样，训练不下去） image