Merge pull request #28 from mysablehats/master

Fixes inconsistent fps issues (fps reading vs fps writing)
ORB-HD · Jan 10, 2023 · 0d2eba8 · zalayeta88 · Feb 11, 2023 · 0d2eba8
2 parents 852286e + d013a08
commit 0d2eba8
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/deface/deface.py b/deface/deface.py
@@ -114,7 +114,11 @@ def video_detect(
         replaceimg = None
 ):
     try:
-        reader: imageio.plugins.ffmpeg.FfmpegFormat.Reader = imageio.get_reader(ipath)
+        if 'fps' in ffmpeg_config:
+            reader: imageio.plugins.ffmpeg.FfmpegFormat.Reader = imageio.get_reader(ipath, fps=ffmpeg_config['fps'])
+        else:
+            reader: imageio.plugins.ffmpeg.FfmpegFormat.Reader = imageio.get_reader(ipath)
+
         meta = reader.get_meta_data()
         _ = meta['size']
     except: