pytorch单张图片预测,如何代替DataLoader

1. 批量读取

trainset, valset, testset = get_dataset(cfg)

test_loader = DataLoader(testset, batch_size=1, shuffle=False, num_workers=cfg['num_workers'])

这是一个 PyTor

你可能感兴趣的:(神经网络,图像处理,pytorch,深度学习,人工智能)