Skip to content

Commit 1fe1e11

Browse files
authored
enable impors (#2918)
1 parent 75e4a7d commit 1fe1e11

File tree

1 file changed

+12
-4
lines changed

1 file changed

+12
-4
lines changed

torchvision/io/__init__.py

+12-4
Original file line numberDiff line numberDiff line change
@@ -17,11 +17,15 @@
1717
write_video,
1818
)
1919
from .image import (
20-
read_image,
2120
decode_image,
21+
decode_jpeg,
22+
decode_png,
2223
encode_jpeg,
23-
write_jpeg,
2424
encode_png,
25+
read_file,
26+
read_image,
27+
write_file,
28+
write_jpeg,
2529
write_png,
2630
)
2731

@@ -164,11 +168,15 @@ def set_current_stream(self, stream: str):
164168
"_read_video_meta_data",
165169
"VideoMetaData",
166170
"Timebase",
167-
"read_image",
168171
"decode_image",
172+
"decode_jpeg",
173+
"decode_png",
169174
"encode_jpeg",
170-
"write_jpeg",
171175
"encode_png",
176+
"read_file",
177+
"read_image",
178+
"write_file",
179+
"write_jpeg",
172180
"write_png",
173181
"Video",
174182
]

0 commit comments

Comments
 (0)