环境:
(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] 按需要更改
[可视化部分组件还在研究怎么用]