跑代码的时候遇到了这种问题,如下图或者代码段中展示
AttributeError: 'str' object has no attribute 'seek'
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/amax/wj/MA-GAN-main/dataset.py", line 54, in __getitem__
input_image = load_img(self.image_filenames[index])
File "/home/amax/wj/MA-GAN-main/dataset.py", line 40, in load_img
img = Image.open(filepath)
File "/home/amax/anaconda3/envs/wujie/lib/python3.8/site-packages/PIL/Image.py", line 3137, in open
fp = io.BytesIO(fp.read())
明显问题出在了数据读取的时候。我查了网上的很多方法,比如说seek()是python处理文本文件的一个函数,用于移动文件读取指针到指定位置,而这个报错的原因是传给seek()的参数不是一个File类,而是一个字符串,那就去一段一段查哪里将文件变成了str,最终无果,那我只能继续看代码了。
解决方法
我发现这段代码先会读取训练数据然后训练好再读取测试数据进行测试,每次都是训练跑完读取测试数据出错,同一段代码读取数据训练集没问题,那就是哪里配置出了问题呗。
training_data_loader = DataLoader(dataset=train_set, batch_size=cfg["train_batch_size"], shuffle=True, drop_last=True)
testing_data_loader = DataLoader(dataset=test_set, batch_size=cfg["test_batch_size"], shuffle=False)
训练和测试的代码只有shuffle的区别,我的测试集shuffle值是false原来,所以只要我选择打乱数据集就能运行了,具体原因还是待探究啊,好歹解决了,实在要不打乱的建议换一个读取数据的代码吧。
文章出处登录后可见!
已经登录?立即刷新