美文网首页目标检测
在COCO 数据集上训练YOLOv3模型

在COCO 数据集上训练YOLOv3模型

作者: LabVIEW_Python | 来源:发表于2019-11-05 12:23 被阅读0次

           前面介绍了《从零开始在Windows10中编译安装YOLOv3》和《在Pascal VOC 数据集上训练YOLOv3》,本节介绍在COCO 数据集上训练YOLOv3。

    第一步,下载并配置COCO数据集。

    a.git clone https://github.com/pdollar/coco

    b. 在coco文件夹里面新建一个images文件夹

    c. 将COCO数据集:

    http://images.cocodataset.org/zips/train2014.zip

    http://images.cocodataset.org/zips/val2014.zip

    下载到images文件夹中,然后解压到当前文件夹

    d. 将COCO Metadata:

    https://pjreddie.com/media/files/instances_train-val2014.zip

    https://pjreddie.com/media/files/coco/5k.part

    https://pjreddie.com/media/files/coco/trainvalno5k.part

    https://pjreddie.com/media/files/coco/labels.tgz

    下载到coco文件夹

    然后将labels.tgz解压到当前文件夹得到labels文件夹

    将instances_train-val2014.zip解压到当前文件夹得到annotations文件夹

    f.在Windows下安装Linux子系统

    参考《在Windows中安装Ubuntu子系统

    g. 在Windows下启动Ubuntu,先用命令

    cd /mnt/d/darknet/coco/

    切换到D盘下的coco文件夹,然后执行命令

    paste <(awk "{print \"$PWD\"}" <5k.part) 5k.part | tr -d '\t' > 5k.txt

    paste <(awk "{print \"$PWD\"}" <trainvalno5k.part) trainvalno5k.part | tr -d '\t' > trainvalno5k.txt

    设置图片列表,如下图所示

    到此,完成在COCO 数据集上训练YOLOv3的COCO数据集下载与配置。

    第二步,修改 coco.data配置文件。

    在D:\darknet\cfg文件夹,打开coco.data,根据实际情况,修改配置文件

    第三步,修改yolov3.cfg 配置文件。

    在D:\darknet\cfg文件夹,打开yolov3.cfg,修改batch 和 subdivisions如下

    第四步,训练模型。

    在D:\darknet文件夹的地址栏中键入“cmd”,启动Windows命令行终端,运行命令:

    darknet.exe detector train cfg\coco.data cfg\yolov3.cfg darknet53.conv.74

    完毕!~

    在自己的数据集上训练YOLOv3模型

    参考文件:

    1, https://pjreddie.com/darknet/yolo/

    参考阅读《深度学习图像识别技术

    相关文章

      网友评论

        本文标题:在COCO 数据集上训练YOLOv3模型

        本文链接:https://www.haomeiwen.com/subject/mcafbctx.html