From 3525d94cdbf3412bbb9b77199890b4f9d4992f20 Mon Sep 17 00:00:00 2001 From: zyan3 Date: Mon, 7 Oct 2019 11:45:48 -0700 Subject: [PATCH] expose more io api --- torchvision/io/__init__.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/torchvision/io/__init__.py b/torchvision/io/__init__.py index 36be33884b8..13958fb9ab4 100644 --- a/torchvision/io/__init__.py +++ b/torchvision/io/__init__.py @@ -1,8 +1,16 @@ from .video import write_video, read_video, read_video_timestamps -from ._video_opt import _read_video_from_file, _read_video_timestamps_from_file, _HAS_VIDEO_OPT +from ._video_opt import ( + _read_video_from_file, + _read_video_timestamps_from_file, + _read_video_from_memory, + _read_video_timestamps_from_memory, + _HAS_VIDEO_OPT, +) __all__ = [ 'write_video', 'read_video', 'read_video_timestamps', - '_read_video_from_file', '_read_video_timestamps_from_file', '_HAS_VIDEO_OPT', + '_read_video_from_file', '_read_video_timestamps_from_file', + '_read_video_from_memory', '_read_video_timestamps_from_memory', + '_HAS_VIDEO_OPT', ]