darknet yolov3 训练自定义数据集

VOC数据集上的训练步骤

修改类别

1. yolo层的classes修改成新的类别书
2. yolo层前一个convolution层的filters修改成(classes+4+1)*3
3. 修改data文件中的类别数目

训练

  ./darknet detector train cfg/custom.data cfg/yolov3.cfg pretrained/darknet53.conv.74 2>&1 train.log

burn_in的效果比较好

测试

单幅图测试

   ./darknet detector test examples/traffic/image.data examples/traffic/yolov3.cfg examples/traffic/yolov3_900.weights data/person_walk.jpg             
   参数说明
   image.data:  训练时指定类别数目，训练/测试图像路径
   yolov3.cfg： 网络定义文件
   yolov3_900.weights: 训练出的网络权重
   person_walk.jpg: 测试图片                                                                        

批量统计recalling

默认darknet的recall模式只能测试coco_val_5k，要测试其他数据集，只需要修改darkne