yololite训练

https://github.com/reu2018DL/YOLO-LITE

yololite训练遇到的问题:

1. 产生anchors:
AlexeyAB的darknet框架下产生anchors:
由于yololite属于yolov2系列,因此使用
./darknet detector calc_anchors cfg/myown.data -num_of_clusters 5 -width 13 -height 13 -show
其中num_of_clusters为cfg文件中anchors的数量对个数, width与height是根据输入图片分辨率经过yololite网络在region层前得到的分辨率。

2. 训练命令
./darknet detector train cfg/myown.data cfg/tiny-yolov2-trial3-noBatch.cfg

在训练的过程中,查看了yolov2的训练过程,若训练yolov2系列,如yolov2-voc.cfg, yolov2-tiny-voc.cfg, yolo-voc.cfg, yolo-voc.2.0.cfg, …,需要下载darknet19_448.conv.23网络(http://pjreddie.com/media/files/darknet19_448.conv.23),放在相应位置,如pjreddie版本的darknet需要放在darknet文件夹下,AlexeyAB版本需要放在build\darknet\x64文件夹下。

因此我最初模仿yolov3训练命令,在终端写入命令:
./darknet detector train cfg/myown.data cfg/tiny-yolov2-trial3-noBatch.cfg darknet19_448.conv.23
训练时出现全是nan的情况。
若删掉后面的darknet19_448.conv.23,则可以正常训练。

AlexeyAB版本darknet下yolov2训练:
https://github.com/AlexeyAB/darknet/tree/47c7af1cea5bbdedf1184963355e6418cb8b1b4f#how-to-train-pascal-voc-data

AlexeyAB版本darknet下yolov3以及yolov4训练:
https://github.com/AlexeyAB/darknet#how-to-train-to-detect-your-custom-objects

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值