pytorch训练自己的数据集

作者：2023面试高手 | 2024-03-18 10:40:34

踩

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)
1
2
3
4
5

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

import os
data = []
labels = []
filetype = "ppm"
for root,dirs,files in os.walk("./"):
    for f in files:
        if f.split('.')[1] == filetype:
            data.append1
2
3
4
5
6
7

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/2023面试高手/article/detail/262450