add read from memory with constructor overload

Author: jdsgomes

setup.py

@@ -351,8 +351,8 @@ def get_extensions():
     # FIXME: causes crash. See the following GitHub issues for more details.
     # FIXME: https://github.com/pytorch/pytorch/issues/65000
     # FIXME: https://github.com/pytorch/vision/issues/3367
-    if sys.platform != "linux" or (sys.version_info.major == 3 and sys.version_info.minor == 9):
-        has_ffmpeg = False
+    # if sys.platform != "linux" or (sys.version_info.major == 3 and sys.version_info.minor == 9):
+    #   has_ffmpeg = False
     if has_ffmpeg:
         try:
             # This is to check if ffmpeg is installed properly.

torchvision/csrc/io/video/video.cpp

@@ -156,7 +156,7 @@ void Video::_getDecoderParams(
 
 } // _get decoder params
 
-Video::Video(std::string videoPath, std::string stream, int64_t numThreads) {
+void Video::_init(std::string stream, int64_t numThreads) {
   C10_LOG_API_USAGE_ONCE("torchvision.csrc.io.video.video.Video");
   // set number of threads global
   numThreads_ = numThreads;
@@ -173,13 +173,6 @@ Video::Video(std::string videoPath, std::string stream, int64_t numThreads) {
       numThreads_ // global number of Threads for decoding
   );
 
-  std::string logMessage, logType;
-
-  // TODO: add read from memory option
-  params.uri = videoPath;
-  logType = "file";
-  logMessage = videoPath;
-
   // locals
   std::vector<double> audioFPS, videoFPS;
   std::vector<double> audioDuration, videoDuration, ccDuration, subsDuration;
@@ -232,7 +225,18 @@ Video::Video(std::string videoPath, std::string stream, int64_t numThreads) {
         << "Stream index set to " << std::get<1>(current_stream)
         << ". If you encounter trouble, consider switching it to automatic stream discovery. \n";
   }
-} // video
+} // Video::Init
+
+
+Video::Video(torch::Tensor videoData, std::string stream, int64_t numThreads) {
+  callback = MemoryBuffer::getCallback(videoData.data_ptr<uint8_t>(), videoData.size(0));
+  Video::_init(stream, numThreads);
+}
+  
+Video::Video(std::string videoPath, std::string stream, int64_t numThreads) {
+  params.uri = videoPath;
+  Video::_init(stream, numThreads);
+}
 
 bool Video::setCurrentStream(std::string stream = "video") {
   if ((!stream.empty()) && (_parseStream(stream) != current_stream)) {

torchvision/csrc/io/video/video.h

@@ -20,6 +20,8 @@ struct Video : torch::CustomClassHolder {
 
  public:
   Video(std::string videoPath, std::string stream, int64_t numThreads);
+  Video(torch::Tensor videoData, std::string stream, int64_t numThreads);
+
   std::tuple<std::string, int64_t> getCurrentStream() const;
   c10::Dict<std::string, c10::Dict<std::string, std::vector<double>>>
   getStreamMetadata() const;
@@ -34,6 +36,8 @@ struct Video : torch::CustomClassHolder {
   // time in comination with any_frame settings
   double seekTS = -1;
 
+  void _init(std::string stream, int64_t numThreads);
+
   void _getDecoderParams(
       double videoStartS,
       int64_t getPtsOnly,