pytorch 数据处理模块
PyTorch 的数据处理模块是 torchvision.transform,只对 PIL.Image 或维度为 (H, W, C) 的图片数据进行数据预处理。由于 OpenCV 读入图片的数据维度是 (H, W, C),所以不能直接使用 torchvision.transform 处理 OpenCV 的图片数据。
1. class torchvision.transforms.Compose(t...
原创
2019-01-22 11:20:23 ·
295 阅读 ·
0 评论