From f202440717b5cfc97c2442425c83bcfbe459aaac Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Wed, 27 Jan 2021 18:23:20 +0000 Subject: [PATCH 1/5] Moving registration of video methods in Video.cpp and removing unnecessary includes. --- torchvision/csrc/io/video/Video.cpp | 16 +++++++++++----- torchvision/csrc/io/video/Video.h | 18 +++++------------- torchvision/csrc/io/video/register.cpp | 14 -------------- .../csrc/io/video_reader/VideoReader.cpp | 12 ++++-------- torchvision/csrc/io/video_reader/VideoReader.h | 4 +++- 5 files changed, 23 insertions(+), 41 deletions(-) delete mode 100644 torchvision/csrc/io/video/register.cpp diff --git a/torchvision/csrc/io/video/Video.cpp b/torchvision/csrc/io/video/Video.cpp index c1e19e0b5f5..7f494a47868 100644 --- a/torchvision/csrc/io/video/Video.cpp +++ b/torchvision/csrc/io/video/Video.cpp @@ -1,9 +1,6 @@ #include "Video.h" -#include -#include -#include "defs.h" -#include "memory_buffer.h" -#include "sync_decoder.h" + +#include using namespace std; using namespace ffmpeg; @@ -319,3 +316,12 @@ std::tuple Video::Next() { return std::make_tuple(outFrame, frame_pts_s); } + +static auto registerVideo = + torch::class_