TypeError: ‘NoneType‘ object is not subscriptable

最新推荐文章于 2024-08-23 08:00:00 发布

fishdreamS233

最新推荐文章于 2024-08-23 08:00:00 发布

阅读量6.7k

点赞数 2

文章标签：神经网络 pytorch 深度学习

本文链接：https://blog.csdn.net/weixin_45677770/article/details/109649264

版权

在使用pandas加载CSV文件时遇到TypeError: 'NoneType' object is not subscriptable的错误。原因是CSV文件头部包含了未定义的head。解决方案是删除数组的第一行，确保数据加载正确。

摘要由CSDN通过智能技术生成

TypeError: 'NoneType' object is not subscriptable -pytorch dataloader 加载image 和 label的解决方法

用pandas读取数据存储为csv格式，在用pytorch的torch.utils.data的Dataset存数据的时候：

class LaneDataset(Dataset): 

    
    def __init__(self, csv_file, transform=None):
        super(LaneDataset, self).__init__()
        self.data = pd.read_csv(os.path.join(os.getcwd(), "data_list", csv_file), header=None,
                                  names=["image",
                                         "label"])

        
        self.images = self.data["image"].values
        self.labels = self.data["label"].values
        self.transform = transform

   
    def