YOLOv5_OBB_KLD

YOLOv5_OBB_KLD copied to clipboard

KLD训练得到权重后，在检测时发现角度信息相差不大

45

KLD训练得到权重后，在检测时发现角度信息相差不大，请问是标签或者是其他地方需要调整吗。 这是三个不同图片检测出来的结果，经过换算后，他们的角度信息几乎一致，检测结果看起来像是水平框检测 ((876.5000610351562, 768.5), (462.3126525878906, 404.125), -87.8736572265625) ((862.5, 677.5000610351562), (275.181640625, 507.555908203125), -87.9666748046875) ((696.5, 1057.5001220703125), (185.2694854736328, 233.78030395507812), -87.05175018310547)

wang-TJ-20

你好，请问一下你跑dota数据集有没有使用多gpu训练

xing-bing

你好，请问你在训练过程中出现精度振荡范围大的情况吗

xing-bing

训练模型预测过程中，预测框问题

7

训练出来的模型在预测过程中，预测结果为这样，想问问作者有没有遇到这样的情况 

whiteNV4

数据集预处理

2

你好 我按照流程切割DOTA1.5的数据集不能用于训练，大佬数据集处理的代码能看一下吗

Ace-chenyang

关于KLD损失的疑问

10

作者你好，感谢你分享的工作，在阅读源码时有几个问题。 您给出的KLD损失绘图如下，显示分类损失一直为0，而角度损失一直下降。  但是，我在阅读compute_loss_KLD中，langle 初始值为0 ，  而后，langle 未参与其他计算，就到达最后损失计算，langle *= h['angle'] *S，所以 langle=0，即 langle 一直为0，角度仅在Box损失计算中参与Iou的计算。如下图  我试着训练DOTA数据集，效果不好。想知道您给的数据集label是什么格式（poly的4点坐标，还是由poly经过CV外接最小矩形的带角度的坐标，由OpenCV法转化为长边法的代码在哪里喃），谢谢 PS，这是我用CSL测全部16类后，得到的ship:0.8941434649109584, 但是现在KLD得到效果不好

LUO77123

new question

3

大佬您好，这个KLD是损失函数的计算吧，在那个yaml文件关于文件的定义中，不用体现出来嘛

Allen-Scai

detect报错

7

大佬大佬，请问要怎么样解决？ 使用TPH训练，使用检测命令加载该训练权重时 python detect.py --weights runs_swin/exp/weights/best.pt 出现报错 Fusing layers... Traceback (most recent call last): File "detect.py", line 350, in detect(kld_flag=opt.kld) File "detect.py", line 54, in detect model = attempt_load(weights, map_location=device)...

Pearliz

置信度很低

5

我发现结果的置信度很低，都是0.00几

Yanhui-He

大佬，训练DOTA1.5单类ship时为啥检测效果这么差啊

2

  detect.py 检测效果只有0.02？大佬麻烦看下

yanxioa