说明:
图像分类相关模型训练
常用命令:
训练脚本
python train_model.py --dataset gender_stars_gen_align --backbone resnet18 --expFolder test0726 --expID 2
- dataset:数据集名称,src/config.py文件的datasets参数中指定数据集;模型保存在weights/{expFolder}/{expID}
- backbone:backbone网络;src/model.py文件中定义各网络结构
配置文件
src/config.py 指定datasets参数