Custom Dataset Training

[hulinlang]

[hulinlang]

P,R一直为0，也预测不出框。

[wanghui589]

我用的那个SSDD_tiny公共数据集也一直没有精度，很难顶。 请问你解决了么·？

[hukaixuan19970627]

https://github.com/hukaixuan19970627/yolov5_obb/issues/386#issuecomment-1186461244 按照这个回答重新发一次哦，提供的信息不大详细。

[hulinlang]

#386 (comment) 按照这个回答重新发一次哦，提供的信息不大详细。

result opt hyp train_batch val_batch_labels val_batch_pre 没有labels的那两个文件。 关于数据集 标注如下 其中有一些带负数的标注，运行时报警告舍弃了。 图像如下 全为1024*1024的SAR图像。 总体一千张，我按9：1分的训练和验证。

[hukaixuan19970627]

没生成labels.jpg那两个文件的话说明该数据集的标注文件不符合格式要求。 你应该试过getstart.md里面的demo吧，正常情况下生成的文件应该和demo保持一致，你首先得排除一下labels对训练的影响，因为你的results.jpg中显示出现过nan值，说明数据可能比较“脏”。 labels有负数的话你加个预处理操作，利用np.clip函数把poly值的最小值限制为0就行。

[hulinlang]

getstart里的demo运行train.py也没有labels.jpg的两个文件，权重文件时下载的您上传的度盘文件，只是名字没改，其他都一样。 运行的时候有个这个警告：UserWarning: Detected call of lr_scheduler.step() before optimizer.step()， 效果如下，感觉也有点问题，我做训练时nc的类型都注意改了的，唯一有改动的是imgsize，我改到了1280. train_batch val_batch_labels val_batch_pre

[hukaixuan19970627]

demo也没有的话，检查下环境依赖的版本和requirements.txt是否保持一致

[hulinlang]

demo也没有的话，检查下环境依赖的版本和requirements.txt是否保持一致 我解决了训练P，R为0的问题，但还是没有labels的两个文件，大概原因就是16xx系的显卡不支持cuda11.3，换回10.2就可以了 具体可以参考这个 https://blog.csdn.net/qq_52902342/article/details/124261371

[Rorsc4achhh]

兄弟，能分享一下作者的weights吗，我能找到的链接都过期了