工作上好难。所有人都在给我压力。
是真的所有人。
我还很难排解,就很抑郁有点。
只能说孽力回馈,上半年没做的事,下半年还要接着做,而且时间更紧。。。
英语,二建,BIM。。。。
我找了三个数据集,DOTA,DIOR,FAIR1M1.0
以后可能用到FAIR1M2.0吧。
DOTA数据集是YOLO格式,网上全是VOC格式转YOLO格式,但是没有教程教我YOLOv3怎么直接用YOLO格式。
我改了两天,发现可能还是pytorch和python水平不太够。。。或者参考的代码太厉害了。
无奈之下,我只能用的VOC格式的DIOR数据集跑的。
22.11.101.png
冻结主干网络的情况下,7分钟一次,50次是350分钟,6个小时,早上起来发现炸显存了。
纵使你是3090ti,还是逃不过炸显存的结局。
不过训练了50次,还是能检测一些图片。
22.11.102.jpg
22.11.103.jpg
22.11.104.jpg
22.11.105.jpg
后来我发现了炸显存的原因,分辨率800*800,冻结网络时batchsize=64,占20G显存,解冻之后batchsize=16占20G显存,一开始我的解冻后batchsize没调整,一解冻直接炸。
训练的时候,开预测也会炸,预测一开就占2G,输入图片瞬间会黑屏,然后训练程序炸显存。。
算是初探目标检测吧。
冻结网络时7-8分钟一次,解冻10分钟一次,300次训练大概是50个小时。
如果按显卡300w,一次训练,用十几度电。。
接下来就是,试试这个网络的KV260部署。
训练YOLOV5并优化(多看论文的优化方法)
部署kv260
训练yolov7并优化
部署zcu102
目前没人做过。
加速吧!!
还有好多事要做!


本文由 sdh 创作,采用 知识共享署名 3.0,可自由转载、引用,但需署名作者且注明文章出处。

还不快抢沙发

添加新评论