tensorflow - Yolo没有开始训练

Question

我正在尝试在自定义数据集上训练 Yolo，一切似乎都没有错误，但这不是训练。

我遵循了 https://github.com/AlexeyAB/darknet 上的教程两次但我得到相同的结果

./darknet detector train data/obj.data cfg/yolo-obj.cfg yolov4.conv.137

[yolo] params: iou loss: ciou (4), iou_norm: 0.07, cls_norm: 1.00, scale_x_y: 1.05
nms_kind: greedynms (1), beta = 0.600000 
Total BFLOPS 59.563 
avg_outputs = 489778 
Loading weights from yolov4.conv.137...
 seen 64, trained: 0 K-images (0 Kilo-batches_64) 
Done! Loaded 137 layers from weights-file 
Learning Rate: 0.001, Momentum: 0.949, Decay: 0.0005
Resizing, random_coef = 1.40 

 608 x 608 
 Create 64 permanent cpu-threads 

 mosaic=1 - compile Darknet with OpenCV for using mosaic=1 

我也试过没有预先训练的权重，但这也没有开始训练过程

./darknet detector train data/obj.data cfg/yolo-obj.cfg
[yolo] params: iou loss: ciou (4), iou_norm: 0.07, cls_norm: 1.00, scale_x_y: 1.05
nms_kind: greedynms (1), beta = 0.600000 
Total BFLOPS 59.563 
avg_outputs = 489778 
Learning Rate: 0.001, Momentum: 0.949, Decay: 0.0005
Resizing, random_coef = 1.40 

 608 x 608 
 Create 64 permanent cpu-threads 

 mosaic=1 - compile Darknet with OpenCV for using mosaic=1 

我错过了什么？

Answer 1

如果要使用OpenCV需要重新编译Darknet，但首先将make文件改成如下:

 OPENCV=1

如果您不需要 OpenCV，请按照 @TaQuangTu 建议的方式进行操作。当您修复此行时，只需再次运行 build.sh 脚本，它应该可以正常工作。
如果您打算使用 GPU 进行训练，我还建议更改以下几行

GPU=1
CUDNN=1
CUDNN_HALF=1