File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -41,8 +41,8 @@ def smoke_test_torchvision() -> None:
41
41
print ('Is torchvision useable?' , all (x is not None for x in [torch .ops .image .decode_png , torch .ops .torchvision .roi_align ]))
42
42
43
43
def smoke_test_torchvision_read_decode () -> None :
44
- from torchvision .io import read_image , decode_jpeg , decode_png
45
- img_jpg = read_image (str (pathlib .Path (__file__ ).parent / 'assets' / 'rgb_pytorch.jpg' ))
44
+ from torchvision .io import read_file , decode_jpeg , decode_png
45
+ img_jpg = read_file (str (pathlib .Path (__file__ ).parent / 'assets' / 'rgb_pytorch.jpg' ))
46
46
img_jpg_nv = decode_jpeg (img_jpg )
47
47
img_png = read_image (str (pathlib .Path (__file__ ).parent / 'assets' / 'rgb_pytorch.png' ))
48
48
assert img_png .ndim == 3 and img_png .numel () > 100
You can’t perform that action at this time.
0 commit comments