expose more io api (#1423)

e48b9584 · Zhicheng Yan · Francisco Massa · 1258bb7b · e48b9584
Commit e48b9584 authored Oct 08, 2019 by Zhicheng Yan Committed by Francisco Massa Oct 08, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 10 additions and 2 deletions

torchvision/io/__init__.py torchvision/io/__init__.py +10 -2

No files found.
--- a/torchvision/io/__init__.py
+++ b/torchvision/io/__init__.py
 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',
 ]