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

你可能感兴趣的:(人工智能,目标检测,计算机视觉)