exu/ffmpeg-encoding-benchmark
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

WIP move encoding to separate functions

Browse Source
This commit is contained in:
RealStickman 2022-10-24 10:08:30 +02:00
parent 828f2a275d
commit f07ccc0efd
1 changed files with 63 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

75
Sparks_in_Blender_encode.py
View File

@ -94,6 +94,54 @@ def write_line(
)
def encode_general(inputfile: str, outputfile: str, codec: str, crf: str, preset: str):
"""
General encoding function
Parameters:
inputfile (str): Path to input file
outputfile (str): Path to output file
codec (str): Codec used
crf (str): CRF value
preset (str): Choosen preset
"""
ff = ffmpy.FFmpeg(
inputs={inputfile: None},
outputs={
outputfile: "-c:v {videocodec} -crf {crf} -preset {preset} -g 240 -map 0:v:0 ".format(
videocodec=codec,
crf=crf,
preset=preset,
)
},
)
return ff
def encode_libaom(inputfile: str, outputfile: str, crf: str, preset: str):
"""
Encoding with libaom
Parameters:
inputfile (str): Path to input file
outputfile (str): Path to output file
crf (str): CRF value
preset (str): Choosen preset
"""
ff = ffmpy.FFmpeg(
inputs={inputfile: None},
outputs={
outputfile: "-c:v libaom-av1 -crf {crf} -b:v 0 -cpu-used {preset} -row-mt 1 -tiles 2x2 -g 240 -map 0:v:0 ".format(
crf=crf,
preset=preset,
)
},
)
return ff
if __name__ == "__main__":
if not os.path.isdir("encoded"):
os.mkdir("encoded")
@ -115,6 +163,8 @@ if __name__ == "__main__":
)
)
inputfile = "Sparks_in_Blender.webm"
for codec in encoding:
for crf in encoding[codec]["crf"]:
for preset in encoding[codec]["presets"]:
@ -133,27 +183,20 @@ if __name__ == "__main__":
# libaom needs additional options
if codec == "libaom-av1":
ff = ffmpy.FFmpeg(
inputs={"Sparks_in_Blender.webm": None},
outputs={
outputfile: "-c:v {videocodec} -crf {crf} -b:v 0 -cpu-used {preset} -row-mt 1 -tiles 2x2 -g 240 -map 0:v:0 ".format(
videocodec=codec,
ff = encode_libaom(
inputfile=inputfile,
outputfile=outputfile,
crf=crf,
preset=preset,
)
},
)
else:
ff = ffmpy.FFmpeg(
inputs={"Sparks_in_Blender.webm": None},
outputs={
outputfile: "-c:v {videocodec} -crf {crf} -preset {preset} -g 240 -map 0:v:0 ".format(
videocodec=codec,
ff = encode_general(
inputfile=inputfile,
outputfile=outputfile,
codec=codec,
crf=crf,
preset=preset,
)
},
)
starttime = now()
ff.run()
@ -162,7 +205,7 @@ if __name__ == "__main__":
outputfilesize = os.path.getsize(outputfile) / 1024 / 1024
ffvmaf = ffmpy.FFmpeg(
ff = ffmpy.FFmpeg(
inputs=OrderedDict(
[(outputfile, None), ("Sparks_in_Blender.webm", None)]
),
@ -173,7 +216,7 @@ if __name__ == "__main__":
},
)
ffvmaf.run()
ff.run()
with open("vmaf.json", "r") as file:
vmaf = json.load(file)