Mobilenet-mxnet配置,数据集制作及训练

环境:
(conda list查看版本号)
win10
cuda10.0
python 3.5
mxnet-cu100 1.5.0(要与cuda版本匹配)
mxnet 1.6.0


数据集制作:
prepare_dataset.py:

  • –set [string] 要用到的.txt文件名(txt文件就是voc中的train)
  • –target [path] 输出.lst文件的位置
  • –class-names [string,string]

./dataset/pascal_voc.py:
line35: ’ .jpg ’

在命令行里运行似乎会有问题,直接修改default在编译器运行。


训练:

要修改的参数:

train.py:

  • –num-class 类别数
  • –resume (这个还没看懂)
  • –class-names “类别名称,逗号隔开”
  • –begin-epoch 1
  • –end-epoch [num]
  • –finetune [num] 如果想要在原有模型基础上接续训练需要更改这个参数
  • –batch-size [num] 按需要更改

[可视化部分组件还在研究怎么用]

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值