opensource/DeOldify
3
0
Fork 0
mirror of https://github.com/jantic/DeOldify.git synced 2026-08-29 02:10:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #114 from mgrankin/fps_fix

Browse Source 
fix audio #113
This commit is contained in:
Jason Antic
2019-05-25 11:29:08 -07:00
committed by GitHub
parent 935b6768d4 96b8ac2974
commit 4c4703e66c
1 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fasterai/visualize.py
+3 -6
View File
@@ -104,13 +104,10 @@ class VideoColorizer():
if re.search('.*?\.jpg', f):
os.remove(os.path.join(dir, f))
def _get_fps(self, source_path: Path)->float:
def _get_fps(self, source_path: Path)->str:
probe = ffmpeg.probe(str(source_path))
stream_data = next((stream for stream in probe['streams'] if stream['codec_type'] == 'video'), None)
avg_frame_rate = stream_data['avg_frame_rate']
fps_num=avg_frame_rate.split("/")[0]
fps_den = avg_frame_rate.rsplit("/")[1]
return round(float(fps_num)/float(fps_den))
return stream_data['avg_frame_rate']
def _download_video_from_url(self, source_url, source_path:Path):
if source_path.exists(): source_path.unlink()
@@ -150,7 +147,7 @@ class VideoColorizer():
if colorized_path.exists(): colorized_path.unlink()
fps = self._get_fps(source_path)
ffmpeg.input(str(colorframes_path_template), format='image2', vcodec='mjpeg', framerate=str(fps)) \
ffmpeg.input(str(colorframes_path_template), format='image2', vcodec='mjpeg', framerate=fps) \
.output(str(colorized_path), crf=17, vcodec='libx264') \
.run(capture_stdout=True)