李宏毅2023HW3

最新推荐文章于 2024-01-22 12:05:50 发布

摸鱼要摸的透彻

最新推荐文章于 2024-01-22 12:05:50 发布

阅读量186

点赞数

文章标签： pytorch

本文链接：https://blog.csdn.net/qq_44016148/article/details/131441818

版权

遇到两种报错

在这里插入图片描述

参考下面的博客

https://blog.csdn.net/loco_monkey/article/details/125565805?spm=1001.2101.3001.6650.7&utm_medium=distribute.pc_relevant.none-task-blog-2%7Edefault%7EBlogCommendFromBaidu%7ERate-7-125565805-blog-124614763.235%5Ev38%5Epc_relevant_anti_t3_base&depth_1-utm_source=distribute.pc_relevant.none-task-blog-2%7Edefault%7EBlogCommendFromBaidu%7ERate-7-125565805-blog-124614763.235%5Ev38%5Epc_relevant_anti_t3_base&utm_relevant_index=13

是拼接路径出了问题
修改dataset
为以下内容

class FoodDataset(Dataset):

    def __init__(self,path,tfm=test_tfm,files = None):
        super(FoodDataset).__init__()
        self.path = path
        self.files = sorted([os.path.join(path,x) for x in os.listdir(path) if x.endswith(".jpg")])
        if files != None:
            self.files = files
        print(f"One {path} sample",self.files[0])
        self.transform = tfm
  
    def __len__(self):
        return len(self.files)
  
    def __getitem__(self,idx):
        fname = self.files[idx]
        im = Image.open(fname)
        im = self.transform(im)
        
        try:
            #label = int(fname.split("/")[-1].split("_")[0])
            #修改为
            label = int(fname.split("\\")[-1].split("_")[0])
        except:
            label = -1 # test has no label
            
        return im,label