使用darknet训练分类器

19 篇文章 0 订阅
6 篇文章 0 订阅

参考:https://pjreddie.com/darknet/train-cifar/

1、在工程目录下新建labels.list,classifier 中train函数路径为了一致也要改动。labels.list内容如下:

n00

n01

n02

2、在cfg文件夹下新建imagenet1k.data,用于指示路径等信息,内容如下:


3、在工程项目目录下建立train文件夹,存放所训练的数据,数据名称包括两部分,第一部分是类别,第二部分是名称的随机取值。

注意:一幅图像名称的第一部分不能是任何图像第一部分的任意子集,也不能出现类别和名称不匹配的情况,否则出现以下错误:


4、撰写网络结构,举例如下:


5、训练

./darknet classifier train cfg/imagenet1k.data cfg/darknet19.cfg

训练状态如下是正常,否则继续找问题:


6、测试

./darknet classifier predict cfg/cifar.data cfg/cifar.cfg cifar.weights 002.png 2

7、验证

./darknet classifier valid cfg/imagenet1k.data cfg/extraction.cfg extraction.weights

使用valid 去前向运算,输出top 1 top 2的值。


注: 最好还是参见官网的分类。

  • 1
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 10
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 10
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值