pytorch训练自己的数据集

使用pytorch自带的模型，并修改全连接层为自己数据集的类别数。

model = models.resnet50(pretrained=False)
class_num = 62
fc_features = model.fc.in_features
model.fc = nn.Linear(fc_features, class_num)
model = model.to(device)

根据数据集文件夹制作包含文件路径 标签的txt文本文件

import os
data = []
labels = []
filetype = "ppm"
for root,dirs,files in os.walk("./"):
    for f in files:
        if f.split('.')[1] == filetype:
            data