test/test_transforms.py · 6fc69a4258aaa85b90bbfb5401fc5cdb8f6daef4 · OpenDAS / vision

fix for to_tensor when input is np.ndarray of shape [H,W,C]. (#55) · 6fc69a42

Alykhan Tejani authored Mar 11, 2017

* fix for to_tensor when input is np.ndarray of shape [H,W,C]. Issue #48 pytorch/vision

* update cifar datasets to transpose images from CHW -> HWC

* fix flake8 issue on test_transforms.py

6fc69a42

test_transforms.py 6.27 KB

Replace test_transforms.py