使用yolov3算法训练自己数据集的测试程序
使用yolov3算法训练自己数据集的测试程序
keras=2.2.4
tensorflow-gpu=1.12.0
cuda=9.0
2020.9.11
记录一下使用yolov3算法的训练过程
9.14
努力使用所学复现一下yolov4,或者说把一些tricks用到yolov3里,试试看效果。
9.17
测试一下yolov4中的一些tricks能否应用到我的项目中,是否会提升一些效率。
9.23
这几天的实验结果都出来了,效果好像还不错
9.28
有时间总结一下前段时间的训练结果
还有就是,tensroflow-gou更新为1.13.1版本,为了匹配cuda10.0,因为要适应一起用电脑的人的配置
10.6
这几天的训练ap值都有些在下降,可能是batch_size因为内存调低的原因下降了
10.8
对于修改网络结构为单一尺度的输出有了更多的想法,但是对于注意力模块这方面好像还没太多的想法,还需要多看看
10.10
确实,对于batchsize的定义会对网络模型有影响,但是还是发现了一个问题,就是10.6提到的精度下降的问题,现在感觉是更换了tensorflow-gpu版本的问题。
换到1.13.1之后的三次训练,结果都不是很好,程序代码方面也没有改动。思考一下这个问题
12.11
重新处理一下图像数据,再梳理一遍训练过程