代码在数据预处理模块中好像存在一些错误:1.一开始就读入全部图像,占用较大内存!问题在mvtec.py里面:
with open(datas_csv_path, 'r') as file:
reader = csv.reader(file)
for row in tqdm(reader):
if row[1] == 'train' and row[0] in CLASS_NAMES:
file_path = os.path.join(root_dir, row[3])
self.paths.append(file_path)
self.x.append(self.transform(Image.open(file_path).convert('RGB')))
代码在数据预处理模块中好像存在一些错误:1.一开始就读入全部图像,占用较大内存!问题在mvtec.py里面: with open(datas_csv_path, 'r') as file: reader = csv.reader(file)
这部分self.x 没必要吧,应该在getitem中调用吧??????