分类：finetuning

2021
08-08

pytorch finetuning 自己的图片进行训练操作

一、pytorchfinetuning自己的图片进行训练这种读取图片的方式用的是torch自带的ImageFolder，读取的文件夹必须在一个大的子文件下，按类别归好类。就像我现在要区分三个类别。#perparedataset#traindatatrain_data=torchvision.datasets.ImageFolder('F:/eyeDataSet/trainData',transform=transforms.Compose([transforms.Scale(256),transforms.CenterCrop(224),transforms.T... 继续阅读 >