美文网首页
Darknet选取VOC的部分类进行训练

Darknet选取VOC的部分类进行训练

作者: myth_0c21 | 来源:发表于2017-09-19 12:09 被阅读0次

    VOC数据集

    总共有20类物体,实际应用中并不需要这么多,可以根据自己的需要,仅训练部分目标

    训练5类目标

    • 更改voc_label.py(位于darknet/script路径下)
    • 将原来的20类更改为你想识别的类
    #classes = ["aeroplane", "bicycle", "bird", "boat", "bottle", "bus", "car", "cat", "chair", "cow", "diningtable", "dog", "horse", "motorbike", "person", "pottedplant", "sheep", "sofa", "train", "tvmonitor"]
    classes = ["bicycle", "bus", "car","motorbike", "person"]
    
    • 执行python voc_label.py
    • 以上步骤,将会把生成label文件,仅标识了你想识别的目标
    • 更改网络结构,开始训练

    相关文章

      网友评论

          本文标题:Darknet选取VOC的部分类进行训练

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