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