From 62672902ec2ff8c216eb1567b9d0fffeec674433 Mon Sep 17 00:00:00 2001
From: Mathieu Beligon <mathieu@feedly.com>
Date: Mon, 9 Mar 2020 21:03:51 -0400
Subject: [PATCH] [common] (frame generators) rename: VideoFrameGenerator ->
FPSVideoFrameGenerator
---
...{video_frame_generator.py => fps_video_frame_generator.py} | 2 +-
.../research_common/dataset/twitch/robots_views_extractor.py | 4 ++--
2 files changed, 3 insertions(+), 3 deletions(-)
rename common/polystar/common/frame_generators/{video_frame_generator.py => fps_video_frame_generator.py} (94%)
diff --git a/common/polystar/common/frame_generators/video_frame_generator.py b/common/polystar/common/frame_generators/fps_video_frame_generator.py
similarity index 94%
rename from common/polystar/common/frame_generators/video_frame_generator.py
rename to common/polystar/common/frame_generators/fps_video_frame_generator.py
index 605228a..280220c 100644
--- a/common/polystar/common/frame_generators/video_frame_generator.py
+++ b/common/polystar/common/frame_generators/fps_video_frame_generator.py
@@ -6,7 +6,7 @@ import ffmpeg
from polystar.common.frame_generators.frames_generator_abc import FrameGeneratorABC
-class VideoFrameGenerator(FrameGeneratorABC):
+class FPSVideoFrameGenerator(FrameGeneratorABC):
def __init__(self, video_path: Path, desired_fps: int):
self.video_path: Path = video_path
self.desired_fps: int = desired_fps
diff --git a/common/research_common/dataset/twitch/robots_views_extractor.py b/common/research_common/dataset/twitch/robots_views_extractor.py
index 878295d..c14759b 100644
--- a/common/research_common/dataset/twitch/robots_views_extractor.py
+++ b/common/research_common/dataset/twitch/robots_views_extractor.py
@@ -3,7 +3,7 @@ import ffmpeg
import numpy as np
from tqdm import tqdm
-from polystar.common.utils.video_frame_generator import VideoFrameGenerator
+from polystar.common.frame_generators.fps_video_frame_generator import FPSVideoFrameGenerator
from research_common.constants import TWITCH_DSET_DIR, TWITCH_ROBOTS_VIEWS_DIR
from research_common.dataset.twitch.mask_detector import is_image_from_robot_view
@@ -15,7 +15,7 @@ class RobotsViewExtractor:
def __init__(self, video_name: str):
self.video_name: str = video_name
self.video_path = TWITCH_DSET_DIR / "videos" / f"{video_name}.mp4"
- self.frame_generator: VideoFrameGenerator = VideoFrameGenerator(self.video_path, self.FPS)
+ self.frame_generator: FPSVideoFrameGenerator = FPSVideoFrameGenerator(self.video_path, self.FPS)
self.count = 0
(TWITCH_ROBOTS_VIEWS_DIR / self.video_name).mkdir(exist_ok=True)
--
GitLab