PR with benchmark results (#410)

pytorch · Nov 23, 2024 · 567243d · 567243d
1 parent 08236be
commit 567243d
Show file tree

Hide file tree

Showing 4 changed files with 348 additions and 156 deletions.
diff --git a/benchmarks/decoders/benchmark_decoders_library.py b/benchmarks/decoders/benchmark_decoders_library.py
@@ -104,7 +104,7 @@ def __init__(self, backend):
 
     def get_frames_from_video(self, video_file, pts_list):
         self.torchvision.set_video_backend(self._backend)
-        reader = self.torchvision.io.VideoReader(video_file, "video")
+        reader = self.torchvision.io.VideoReader(video_file, "video", num_threads=0)
         frames = []
         for pts in pts_list:
             reader.seek(pts)
@@ -114,7 +114,7 @@ def get_frames_from_video(self, video_file, pts_list):
 
     def get_consecutive_frames_from_video(self, video_file, numFramesToDecode):
         self.torchvision.set_video_backend(self._backend)
-        reader = self.torchvision.io.VideoReader(video_file, "video")
+        reader = self.torchvision.io.VideoReader(video_file, "video", num_threads=0)
         frames = []
         for _ in range(numFramesToDecode):
             frame = next(reader)

diff --git a/benchmarks/decoders/benchmark_readme_chart.png b/benchmarks/decoders/benchmark_readme_chart.png