我有一个名为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)
您可以使用
torchvision
库中的transforms
来执行此操作。您可以将声明为参数的任何转换传递到用于创建my_dataset
的任何class
中,如下所示:相关问题 更多 >
编程相关推荐