Ma/video2x
1
0
Fork 0
mirror of https://github.com/k4yt3x/video2x.git synced 2026-02-21 06:05:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

V1.1 alpha: Fully working, added docs

Browse Source
This commit is contained in:
K4YT3X
2018-02-24 22:52:04 -05:00
parent e198972c35
commit 994a542d52
3 changed files with 128 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

61
ffmpeg.py
View File

@@ -9,26 +9,73 @@ Last Modified: Feb 24, 2018
Description: This class handles all FFMPEG related
operations.
Version 1.0
Version 1.1
"""
import os
class FFMPEG:
"""This class communicates with ffmpeg
This class deals with ffmpeg. It handles extracitng
frames, stripping audio, converting images into videos
and inserting audio tracks to videos.
"""
def __init__(self, ffmpeg_path, outfile):
self.ffmpeg_path = ffmpeg_path
self.outfile = outfile
def strip_frames(self, videoin, outpath):
os.system("{} -i {} {}/extracted_%0d.png -y".format(self.ffmpeg_path, videoin, outpath))
def extract_frames(self, videoin, outpath):
"""Extract every frame from original videos
This method extracts every frame from videoin
using ffmpeg
Arguments:
videoin {string} -- input video path
outpath {string} -- video output folder
"""
os.system(
"{} -i {} {}/extracted_%0d.png -y".format(self.ffmpeg_path, videoin, outpath))
def extract_audio(self, videoin, outpath):
os.system("{} -i {} -vn -acodec copy {}/output-audio.aac -y".format(self.ffmpeg_path, videoin, outpath))
"""Strips audio tracks from videos
This method strips audio tracks from videos
into the output folder in aac format.
Arguments:
videoin {string} -- input video path
outpath {string} -- video output folder
"""
os.system(
"{} -i {} -vn -acodec copy {}/output-audio.aac -y".format(self.ffmpeg_path, videoin, outpath))
def to_vid(self, framerate, resolution, folder):
os.system("{} -r {} -f image2 -s {} -i {}/extracted_%d.png -vcodec libx264 -crf 25 -pix_fmt yuv420p output.mp4 -y".format(self.ffmpeg_path, framerate, resolution, folder))
"""Converts images into videos
def pressin_audio(self, videoin, outpath):
os.system("{} -i {} -i {}/output-audio.aac -codec copy -shortest {} -y".format(self.ffmpeg_path, videoin, outpath, self.outfile))
This method converts a set of images into a
video.
Arguments:
framerate {float} -- target video framerate
resolution {string} -- target video resolution
folder {string} -- source images folder
"""
os.system("{} -r {} -f image2 -s {} -i {}/extracted_%d.png -vcodec libx264 -crf 25 -pix_fmt yuv420p output.mp4 -y".format(
self.ffmpeg_path, framerate, resolution, folder))
def insert_audio_track(self, videoin, outpath):
"""Insert audio into video
Inserts the AAC audio track stripped from
the original video into final video.
Arguments:
videoin {string} -- input video path
outpath {string} -- video output folder
"""
os.system("{} -i {} -i {}/output-audio.aac -codec copy -shortest {} -y".format(
self.ffmpeg_path, videoin, outpath, self.outfile))