首先 下载项目完整代码,在pycharm中打开
1. comfig 中有各种网络模型,可以直接使用训练好的预训练模型,尽量不要改动网络模型的结构
2. 18表示网络机构18层,8是每个卡的batch,cifar10 是数据集
3.配置文件解析
4.
类在哪里呢?
4.训练项目
a.首先在config中选择合适的模型,右键 复制配置文件的绝对路径
b. 第二步,打开train.py ,注释处粘贴模型地址,并把\ 变为、\\
C.配置参数
将配置文件的路径复制粘贴在parameters中
D.执行train.py可以训练出来一个模型,保存在
E.rename 配置文件,复制粘贴在config 对应文件夹中,这就是一个完整的配置文件了,可以对该配置文件进行各种自己的操作,也可以修改来训练自己的数据集
F. 在自己的配置文件中修改训练自己的数据集
原来的数据集 train中每个文件夹为一类,共102类
运行filelist.py文件,把train和val文件夹中的文件都复制到train_filelist和val_filelist中,
新的数据文件夹train_filelist中只有花朵,标签在文本文件train.txt中
修改train.py中的参数为自己的配置文件
执行train.py 训练自己的数据集 ,生成新的模型。模型保存路径: