PaddleDetection训练自己的数据集

paddledetection训练自己的数据集

具体步骤

1.数据准备,默认已经安装paddledetection

链接: 多种格式准备链接
以VOC格式为例,格式符合图片要求

2.修改配置文件

vi configs/yolov3/yolov3_mobilenet_v1_270e_voc.yml 

修改以下配置文件

  '../datasets/voc.yml',
  '../runtime.yml',
  '_base_/optimizer_270e.yml',
  '_base_/yolov3_darknet53.yml',
  '_base_/yolov3_reader.yml',

2.1 修改训练类别数和训练集路径

vi /datasets/voc.yml

2.2 修改训练轮次和学习率

  '_base_/optimizer_270e.yml',

2.3 修改batch_size

  '_base_/yolov3_reader.yml',

2.4 开始训练

python tools/train.py -c  configs/yolov3/yolov3_mobilenet_v1_270e_voc.yml 

3.推理预测

 python tools/infer.py -c configs/yolov3/yolov3_mobilenet_v1_270e_voc.yml  \
                    --infer_img=dataset/guoqi_voc/images/6B30964333902A77482020E1B4162B33C7375B6B_size1882_w1268_h713.png \
                    --output_dir=infer_output/ \
                    --draw_threshold=0.3 \
                    -o weights=output/yolov3_mobilenet_v1_270e_voc/model_final \
                    --use_vdl=Ture