62 lines
1.6 KiB
Python
62 lines
1.6 KiB
Python
import subprocess
|
|
import sys
|
|
from pathlib import Path
|
|
import os
|
|
import re
|
|
|
|
output_folder_name ="/bulk_convert_out/"
|
|
|
|
droppedFile = sys.argv[1]
|
|
droppedName = Path(droppedFile).name
|
|
originalfiles = sys.orig_argv[2:]
|
|
|
|
#long path shenanigans
|
|
joined = (' '.join(originalfiles))
|
|
re.sub (' +', ' ', joined)
|
|
listed = re.split(r'(.+?mkv|.+?mov|.+?mp4)', joined)
|
|
listed = list(filter(None, listed))
|
|
listed = [i.lstrip() for i in listed]
|
|
|
|
#print(droppedName)
|
|
|
|
#get path of script location and remove the tail
|
|
head_tail = os.path.split(sys.argv[0])
|
|
output_path = head_tail[0]
|
|
|
|
|
|
#skip first part of sys.argv since it points to script location
|
|
arguments = sys.argv[1:]
|
|
|
|
#check if output folder exists, otherwise create it
|
|
isExist = os.path.exists(output_path+output_folder_name)
|
|
if not isExist:
|
|
os.makedirs(output_path+output_folder_name)
|
|
|
|
|
|
print("input 0: ", listed[0])
|
|
print("input 1: ", listed[1])
|
|
x=int(input("select reference file: "))
|
|
ref= listed.pop(x)
|
|
print(listed)
|
|
print("selected reference file is: ", ref)
|
|
print("1: vmaf_v0.6.1.json")
|
|
print("2: vmaf_4k_v0.6.1.json")
|
|
x=int(input("select vmaf model: "))
|
|
|
|
if x == 1:
|
|
scale = '"scale=1920:1080"'
|
|
model = "vmaf_v0.6.1"
|
|
else:
|
|
scale = '"scale=3840:2160"'
|
|
model = "vmaf_4k_v0.6.1"
|
|
|
|
|
|
#basename = os.path.basename(file)
|
|
|
|
args = 'ffmpeg.exe -i "{distorted}" -i "{reference}" -lavfi "[0:v]{scale}[distorted];[1:v]{scale}[reference]; [distorted][reference]libvmaf='"'"'model=version=vmaf_v0.6.1:feature=name=psnr|name=float_ssim:n_threads=12:log_path=log.txt:log_fmt=json'"'"'" -f null -'.format(distorted=listed[0],reference=ref,scale=scale, model=model)
|
|
print(args)
|
|
subprocess.run(args)
|
|
|
|
|
|
|
|
k=input("press close to exit") |