Pytorch Resnet18 自定义图像数据集分类 训练 1 数据集准备:将数据集按照文件夹分类(类别使用数字,从0开始) 2 修改配置文件config.py 配置好分类总数, 数据集路径等 3 开启训练 python train.py 预测 1 修改配置文件config.py 指定要加载的模型,测试集路径等 2 开启预测 python predict.py tensorboard 查看 1 tensorboard --logdir=runs 运行截图
