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

1. 批量读取

trainset, valset, testset = get_dataset(cfg)

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

这是一个 PyTor

猜你喜欢

转载自blog.csdn.net/qq_26696715/article/details/129649701
今日推荐