小编rah*_*rma的帖子

我有一个名为 的列表wordImages。它包含 np.array 格式的图像，具有不同的宽度和高度。

如何将其转换为张量并使用它而不是my_dataset在下面的代码中？

目前我正在使用这个。但我需要保存/读取图像

demo_data = RawDataset(root="output_craft/", opt=opt) 

demo_loader = torch.utils.data.DataLoader(
                demo_data , batch_size=opt.batch_size,
                shuffle=False,
                num_workers=int(opt.workers),
                collate_fn=AlignCollate_demo, pin_memory=True)