6.训练的时候若数据集小目标较少106层会大概率输出nan,这是数据集的问题如果数据集没有问题,可以通过调大batch或者调小learning_rate来解决(yolo的训练调参略烦)
7.github上暂时还没有发现在pytorch框架下较好用的训练代码,尝试着跑了两个效果不好,所以还是使用darknet来进行训练
10.由于数据集的问题没有小目标,尝试使用三个数据集┅起训练但大幅输出nan表示训练很糟糕。所以在原有第一个数据集训练最后保存的模型的基础上进行后续训练。也就是把预训练权重换荿自己的xxx.weights训练的输出暂时看起来较为正常,等训练完成了看结果好坏(同时需要调整cfg文件的max_batches,比如第一次训练完保存的是100000次生成了final.weights,那么接下去训练就需要把cfg调大超过10w次不然会直接保存。)
后续补充新手入门各种踩坑。