add log file system.
This commit is contained in:
JunHo Kim committed
1 parent
9b0a69fb2b
commit
001a1d9544
1 file changed
+273
-198
+273
-198
@@ -6,12 +6,14 @@ import re
|
|||||||
import subprocess
|
import subprocess
|
||||||
import argparse
|
import argparse
|
||||||
import copy
|
import copy
|
||||||
|
import datetime
|
||||||
import time
|
import time
|
||||||
|
|
||||||
parser = argparse.ArgumentParser()
|
parser = argparse.ArgumentParser()
|
||||||
parser.add_argument("-s", "--source_dir", default="./", help="Input source vid directory")
|
parser.add_argument("-i", "--input_dir", default="./", help="Input source vid directory")
|
||||||
parser.add_argument("-i", "--ivtc_dir", default="./ivtc", help="Input ivtc vid directory")
|
# parser.add_argument("-i", "--ivtc_dir", default="./0729_work/ivtc", help="Input ivtc vid directory")
|
||||||
parser.add_argument("-o", "--output_dir", default="./out", help="Output vid directory")
|
parser.add_argument("-o", "--output_dir", default="./out", help="Output vid directory")
|
||||||
|
parser.add_argument("-m", "--manual_mode", default=False, action="store_true", help="Manual cropping without using log file")
|
||||||
parser.add_argument("-d", "--debug", default=False, action="store_true", help="for debugging. short output")
|
parser.add_argument("-d", "--debug", default=False, action="store_true", help="for debugging. short output")
|
||||||
args = parser.parse_args()
|
args = parser.parse_args()
|
||||||
|
|
||||||
@@ -66,224 +68,280 @@ def resizeCropInfo(key, cropPos, frame_width, frame_height):
|
|||||||
|
|
||||||
return cropPos
|
return cropPos
|
||||||
|
|
||||||
|
def getTime():
|
||||||
|
dt_now = datetime.datetime.now()
|
||||||
|
dt_str = dt_now.strftime('[%Y-%m-%d %H:%M:%S]')
|
||||||
|
# print(dt_str)
|
||||||
|
|
||||||
|
return dt_str
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
vid_types = ('*.avi', '*.mov', '*.mp4', '*.mxf')
|
vid_types = ('*.avi', '*.mov', '*.mp4', '*.mxf')
|
||||||
source_dir = args.source_dir
|
manual_mode = args.manual_mode
|
||||||
ivtc_dir = args.ivtc_dir
|
input_dir = args.input_dir
|
||||||
|
# ivtc_dir = args.ivtc_dir
|
||||||
output_dir = args.output_dir
|
output_dir = args.output_dir
|
||||||
|
|
||||||
source_list = []
|
t = time.time
|
||||||
ivtc_list = []
|
|
||||||
|
input_list = []
|
||||||
|
# ivtc_list = []
|
||||||
for types in vid_types :
|
for types in vid_types :
|
||||||
source_list += sorted(glob.glob(os.path.join(source_dir, types)))
|
input_list += sorted(glob.glob(os.path.join(input_dir, types)))
|
||||||
|
|
||||||
ivtc_list += sorted(glob.glob(os.path.join(ivtc_dir, "*.mp4")))
|
"""
|
||||||
|
Compare "args.source_dir", "args.ivtc_dir"
|
||||||
|
check difference of two input, make ivtc version of source_dir videos.
|
||||||
|
----- NOT USED CODE [07/30] -----
|
||||||
|
"""
|
||||||
|
# ivtc_list += sorted(glob.glob(os.path.join(ivtc_dir, "*.mp4")))
|
||||||
|
|
||||||
if len(ivtc_list) > 0:
|
# if len(ivtc_list) > 0:
|
||||||
ivtc_input_list = []
|
# ivtc_input_list = []
|
||||||
for i, fpath in enumerate(source_list):
|
# for i, fpath in enumerate(source_list):
|
||||||
vid_name = os.path.basename(fpath)[:-4]
|
# vid_name = os.path.basename(fpath)[:-4]
|
||||||
ivtc_name_list = [os.path.basename(x)[:-4] for x in ivtc_list]
|
# ivtc_name_list = [os.path.basename(x)[:-4] for x in ivtc_list]
|
||||||
|
|
||||||
if not vid_name + "_ivtc" in ivtc_name_list :
|
# if not vid_name + "_ivtc" in ivtc_name_list :
|
||||||
ivtc_input_list += glob.glob(fpath)
|
# ivtc_input_list += glob.glob(fpath)
|
||||||
print(f"[INFO]\t Seems \"{os.path.basename(fpath)}\" is stil telecine... add to ivtc processing list")
|
# print(f"[INFO]\t Seems \"{os.path.basename(fpath)}\" is stil telecine... add to ivtc processing list")
|
||||||
else :
|
# else :
|
||||||
ivtc_input_list = source_list
|
# ivtc_input_list = source_list
|
||||||
|
|
||||||
for i, ivtc_input in enumerate(ivtc_input_list) :
|
# for i, ivtc_input in enumerate(ivtc_input_list) :
|
||||||
ivtc_input_num = len(ivtc_input_list)
|
# ivtc_input_num = len(ivtc_input_list)
|
||||||
ivtc_input_name = os.path.basename(ivtc_input)
|
# ivtc_input_name = os.path.basename(ivtc_input)
|
||||||
ivtc_output_path = os.path.join(ivtc_dir, ivtc_input_name[:-4]) + "_ivtc.mp4"
|
# ivtc_output_path = os.path.join(ivtc_dir, ivtc_input_name[:-4]) + "_ivtc.mp4"
|
||||||
|
|
||||||
if not os.path.exists(ivtc_dir) :
|
# if not os.path.exists(ivtc_dir) :
|
||||||
os.mkdir(ivtc_dir)
|
# os.mkdir(ivtc_dir)
|
||||||
print(f'[INFO]\t {ivtc_dir} created')
|
# print(f'[INFO]\t {ivtc_dir} created')
|
||||||
|
|
||||||
print(f"[INFO]\t Work ivtc on \"{ivtc_input_name}\" ===> \"{ivtc_output_path}\" ... [{i + 1}/{ivtc_input_num}]")
|
# print(f"[INFO]\t Work ivtc on \"{ivtc_input_name}\" ===> \"{ivtc_output_path}\" ... [{i + 1}/{ivtc_input_num}]")
|
||||||
print("[INFO]\t Please wait...")
|
# print("[INFO]\t Please wait...")
|
||||||
start_time = time.time()
|
# start_time = time.time()
|
||||||
subprocess.run(["ffmpeg", "-ss", "10", "-i", ivtc_input, "-vf", "fieldmatch=mode=pc_n:combmatch=full,bwdif=0:-1:1,decimate", "-c:v", "libx264", "-crf", "0",
|
# subprocess.run(["ffmpeg", "-ss", "10", "-i", ivtc_input, "-vf", "fieldmatch=mode=pc_n:combmatch=full,bwdif=0:-1:1,decimate", "-c:v", "libx264", "-crf", "0",
|
||||||
"-preset", "ultrafast","-loglevel", "quiet", "-y", ivtc_output_path])
|
# "-preset", "ultrafast","-loglevel", "quiet", "-y", ivtc_output_path])
|
||||||
t = time.time() - start_time
|
# t = time.time() - start_time
|
||||||
print("[INFO]\t Work done. [It takes %.2f seconds.]" % t)
|
# print("[INFO]\t Work done. [It takes %.2f seconds.]" % t)
|
||||||
|
# ivtc_dir = []
|
||||||
|
# ivtc_dir += sorted(glob.glob(os.path.join(ivtc_dir, "*.mp4")))
|
||||||
|
|
||||||
|
# target_ratio = 16 / 9
|
||||||
|
# print("[INFO]\t Target ratio : %.7f" %target_ratio)
|
||||||
|
# print("[INFO]\t Input ivtc dir :", ivtc_dir)
|
||||||
|
|
||||||
ivtc_list = []
|
print("[INFO]\t Input dir :", input_dir)
|
||||||
ivtc_list += sorted(glob.glob(os.path.join(ivtc_dir, "*.mp4")))
|
for i, fpath in enumerate(input_list):
|
||||||
target_ratio = 16 / 9
|
|
||||||
print("[INFO]\t Target ratio : %.7f" %target_ratio)
|
|
||||||
print("[INFO]\t Input ivtc dir :", ivtc_dir)
|
|
||||||
for i, fpath in enumerate(ivtc_list):
|
|
||||||
vid_name = os.path.basename(fpath)[:-4]
|
vid_name = os.path.basename(fpath)[:-4]
|
||||||
vid_num = len(ivtc_list)
|
vid_num = len(input_list)
|
||||||
|
logs = ''
|
||||||
|
|
||||||
# Korean title text bug fix (work in progress)
|
|
||||||
# vid_name = vid_name.encode('utf-8').decode('cp949')
|
|
||||||
|
|
||||||
print(f"[INFO]\t Video num : {i + 1} / {vid_num}")
|
if not manual_mode:
|
||||||
print("[INFO]\t File to detect crop area: ", fpath)
|
if not os.path.exists(f"./logs/{vid_name}.txt"):
|
||||||
print("[INFO]\t Detecting crop area...")
|
print(f"[INFO]\t Can`t find log file. [./logs/{vid_name}.txt]")
|
||||||
p = subprocess.Popen(["ffmpeg", "-ss", "10", "-i", fpath, "-vf", "cropdetect=limit=38",
|
|
||||||
"-vframes", "3000", "-f", "null", "out.null"], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
|
||||||
infos = p.stderr.read().decode('utf-8')
|
|
||||||
# print (infos)
|
|
||||||
allCrops = re.findall("crop=\S+", infos)
|
|
||||||
# print ("Crop Infos : ")
|
|
||||||
# print (allCrops)
|
|
||||||
mostCommonCrop = Counter(allCrops).most_common(1)
|
|
||||||
|
|
||||||
if len(mostCommonCrop) > 0 :
|
|
||||||
crop = mostCommonCrop[0][0]
|
|
||||||
else :
|
|
||||||
print(f"[!!! ERROR !!!]\t Can`t read input vid. Skip to next vid \t ({fpath})")
|
|
||||||
continue
|
|
||||||
|
|
||||||
# Delete "crop=" in list "crop"
|
|
||||||
cropPos = crop.split(":")
|
|
||||||
cropPos[0] = cropPos[0][5:]
|
|
||||||
print(f"[INFO]\t Most common detected crop area: [w : {cropPos[0]}, h : {cropPos[1]}, x : {cropPos[2]}, y : {cropPos[3]}]")
|
|
||||||
cropPos = [int(x) for x in cropPos]
|
|
||||||
|
|
||||||
# Capture vid, Get Vid info
|
|
||||||
capture = cv2.VideoCapture(fpath)
|
|
||||||
frame_width = capture.get(cv2.CAP_PROP_FRAME_WIDTH)
|
|
||||||
frame_height = capture.get(cv2.CAP_PROP_FRAME_HEIGHT)
|
|
||||||
fps = capture.get(cv2.CAP_PROP_FPS)
|
|
||||||
frame_pos = 0
|
|
||||||
frame_count = 0
|
|
||||||
|
|
||||||
# Crop Area even num check
|
|
||||||
if (cropPos[0] % 2) != 0 :
|
|
||||||
if (cropPos[0] + 1) >= frame_width:
|
|
||||||
cropPos[0] -= 1
|
|
||||||
else:
|
|
||||||
cropPos[0] += 1
|
|
||||||
if (cropPos[1] % 2) != 0 :
|
|
||||||
if (cropPos[1] + 1) >= frame_width:
|
|
||||||
cropPos[1] -= 1
|
|
||||||
else:
|
|
||||||
cropPos[1] += 1
|
|
||||||
|
|
||||||
# Crop Area Multiple of 4 check
|
|
||||||
if (cropPos[0] % 4) != 0 :
|
|
||||||
if (cropPos[0] + 2) >= frame_width:
|
|
||||||
cropPos[0] -= 2
|
|
||||||
else:
|
|
||||||
cropPos[0] += 2
|
|
||||||
if (cropPos[1] % 4) != 0 :
|
|
||||||
if (cropPos[1] + 2) >= frame_width:
|
|
||||||
cropPos[1] -= 2
|
|
||||||
else:
|
|
||||||
cropPos[1] += 2
|
|
||||||
|
|
||||||
key = None
|
|
||||||
isPaused = False
|
|
||||||
pausedFrame = None
|
|
||||||
frame = None
|
|
||||||
cv2.namedWindow(f'{vid_name}', cv2.WINDOW_NORMAL | cv2.WINDOW_KEEPRATIO)
|
|
||||||
while True:
|
|
||||||
key = cv2.waitKeyEx(10)
|
|
||||||
|
|
||||||
# Close button(X button) event
|
|
||||||
if cv2.getWindowProperty(f'{vid_name}', cv2.WND_PROP_VISIBLE) < 1 :
|
|
||||||
return
|
|
||||||
|
|
||||||
if key > 0 :
|
|
||||||
# 13 == Enter key ASCII Code
|
|
||||||
if key == 13:
|
|
||||||
break
|
|
||||||
# Pause
|
|
||||||
elif key == ord('p') or key == ord('P'):
|
|
||||||
if isPaused :
|
|
||||||
isPaused = False
|
|
||||||
else :
|
|
||||||
isPaused = True
|
|
||||||
frame = copy.deepcopy(pausedFrame)
|
|
||||||
# Replay
|
|
||||||
elif key == ord('r') or key == ord('R'):
|
|
||||||
if not isPaused:
|
|
||||||
capture = cv2.VideoCapture(fpath)
|
|
||||||
continue
|
|
||||||
# Move Forward (10s)
|
|
||||||
elif key == ord('e') or key == ord('E'):
|
|
||||||
if not isPaused:
|
|
||||||
capture.set(cv2.CAP_PROP_POS_FRAMES, frame_pos + (fps * 10))
|
|
||||||
# Move Backward (10s)
|
|
||||||
elif key == ord('q') or key == ord('Q'):
|
|
||||||
if not isPaused:
|
|
||||||
capture.set(cv2.CAP_PROP_POS_FRAMES, frame_pos - (fps * 10))
|
|
||||||
elif key == ord('h') or key == ord('H'):
|
|
||||||
print("[INFO]\t ### KEY GUIDE ########################################################################################################")
|
|
||||||
print("[INFO]\t # (MOVE : wasd), (RESIZE : ijkl), (FOR / BACKWARD : q, e), (RESET : r), (PAUSE : p), (SKIP : n), (CONFIRM : enter) #")
|
|
||||||
print("[INFO]\t ###################################################################################################################### ")
|
|
||||||
# Skip this Vid
|
|
||||||
elif key == ord('n') or key == ord('N'):
|
|
||||||
break;
|
|
||||||
else:
|
|
||||||
# Crop Area control
|
|
||||||
cropPos = resizeCropInfo(key, cropPos, frame_width, frame_height)
|
|
||||||
|
|
||||||
frame_pos = capture.get(cv2.CAP_PROP_POS_FRAMES)
|
|
||||||
frame_count = capture.get(cv2.CAP_PROP_FRAME_COUNT)
|
|
||||||
|
|
||||||
if(frame_pos >= frame_count):
|
|
||||||
break
|
|
||||||
|
|
||||||
print("[INFO]\t Current crop area : ", end="")
|
|
||||||
print(cropPos, end="")
|
|
||||||
print(f"\t Frame Count ({frame_pos}/{frame_count})\t[PRESS \"h\" TO VIEW KEY GUIDE]", end="\r")
|
|
||||||
|
|
||||||
if not isPaused:
|
|
||||||
ret, frame = capture.read()
|
|
||||||
pausedFrame = copy.deepcopy(frame)
|
|
||||||
cv2.rectangle(frame, (cropPos[2], cropPos[3]), (cropPos[2] + cropPos[0],
|
|
||||||
cropPos[3] + cropPos[1]), (0, 0, 255), thickness=1, lineType=cv2.LINE_8)
|
|
||||||
cv2.imshow(f'{vid_name}', frame)
|
|
||||||
else :
|
else :
|
||||||
pausedFrame = copy.deepcopy(frame)
|
print(f"[INFO]\t Loading log file... [./logs/{vid_name}.txt]")
|
||||||
cv2.rectangle(pausedFrame, (cropPos[2], cropPos[3]), (cropPos[2] + cropPos[0],
|
f = open(f"./logs/{vid_name}.txt", "r")
|
||||||
cropPos[3] + cropPos[1]), (0, 0, 255), thickness=1, lineType=cv2.LINE_8)
|
logs = f.read()
|
||||||
cv2.imshow(f'{vid_name}', pausedFrame)
|
|
||||||
|
|
||||||
print("")
|
if logs is not '':
|
||||||
capture.release()
|
# Delete "crop=" in list "crop"
|
||||||
cv2.destroyAllWindows()
|
allCrops = re.findall("crop=\S+", logs)
|
||||||
|
cropPos = allCrops[-1].split(":")
|
||||||
|
cropPos[0] = cropPos[0][5:]
|
||||||
|
print(f"[INFO]\t Crop area from log file: [w : {cropPos[0]}, h : {cropPos[1]}, x : {cropPos[2]}, y : {cropPos[3]}]")
|
||||||
|
cropPos = [int(x) for x in cropPos]
|
||||||
|
else :
|
||||||
|
print(f"[!!! WARNNIG !!!]\t Empty log file. [./logs/{vid_name}.txt]")
|
||||||
|
|
||||||
# Debugging
|
if (logs is '') or manual_mode:
|
||||||
# cropPos[0] = 678
|
# Korean title text bug fix (work in progress)
|
||||||
# cropPos[1] = 402
|
# vid_name = vid_name.encode('utf-8').decode('cp949')
|
||||||
# cropPos[2] = 20
|
|
||||||
# cropPos[3] = 38
|
print(f"[INFO]\t Video num : {i + 1} / {vid_num}")
|
||||||
# pad_arg = ""
|
print("[INFO]\t File to detect crop area: ", fpath)
|
||||||
|
print("[INFO]\t Detecting crop area...")
|
||||||
|
p = subprocess.Popen(["ffmpeg", "-ss", "10", "-i", fpath, "-vf", "cropdetect=limit=38",
|
||||||
|
"-vframes", "3000", "-f", "null", "out.null"], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
||||||
|
infos = p.stderr.read().decode('utf-8')
|
||||||
|
# print (infos)
|
||||||
|
allCrops = re.findall("crop=\S+", infos)
|
||||||
|
# print ("Crop Infos : ")
|
||||||
|
# print (allCrops)
|
||||||
|
mostCommonCrop = Counter(allCrops).most_common(1)
|
||||||
|
|
||||||
|
if len(mostCommonCrop) > 0 :
|
||||||
|
crop = mostCommonCrop[0][0]
|
||||||
|
else :
|
||||||
|
print(f"[!!! ERROR !!!]\t Can`t read input vid. Skip to next vid \t ({fpath})")
|
||||||
|
continue
|
||||||
|
|
||||||
|
# Delete "crop=" in list "crop"
|
||||||
|
cropPos = crop.split(":")
|
||||||
|
cropPos[0] = cropPos[0][5:]
|
||||||
|
print(f"[INFO]\t Most common detected crop area: [w : {cropPos[0]}, h : {cropPos[1]}, x : {cropPos[2]}, y : {cropPos[3]}]")
|
||||||
|
cropPos = [int(x) for x in cropPos]
|
||||||
|
|
||||||
|
# Capture vid, Get Vid info
|
||||||
|
capture = cv2.VideoCapture(fpath)
|
||||||
|
frame_width = capture.get(cv2.CAP_PROP_FRAME_WIDTH)
|
||||||
|
frame_height = capture.get(cv2.CAP_PROP_FRAME_HEIGHT)
|
||||||
|
fps = capture.get(cv2.CAP_PROP_FPS)
|
||||||
|
frame_pos = 0
|
||||||
|
frame_count = 0
|
||||||
|
|
||||||
|
# Crop Area even num check
|
||||||
|
if (cropPos[0] % 2) != 0 :
|
||||||
|
if (cropPos[0] + 1) >= frame_width:
|
||||||
|
cropPos[0] -= 1
|
||||||
|
else:
|
||||||
|
cropPos[0] += 1
|
||||||
|
if (cropPos[1] % 2) != 0 :
|
||||||
|
if (cropPos[1] + 1) >= frame_width:
|
||||||
|
cropPos[1] -= 1
|
||||||
|
else:
|
||||||
|
cropPos[1] += 1
|
||||||
|
|
||||||
|
# Crop Area Multiple of 4 check
|
||||||
|
if (cropPos[0] % 4) != 0 :
|
||||||
|
if (cropPos[0] + 2) >= frame_width:
|
||||||
|
cropPos[0] -= 2
|
||||||
|
else:
|
||||||
|
cropPos[0] += 2
|
||||||
|
if (cropPos[1] % 4) != 0 :
|
||||||
|
if (cropPos[1] + 2) >= frame_width:
|
||||||
|
cropPos[1] -= 2
|
||||||
|
else:
|
||||||
|
cropPos[1] += 2
|
||||||
|
|
||||||
|
key = None
|
||||||
|
isPaused = False
|
||||||
|
pausedFrame = None
|
||||||
|
frame = None
|
||||||
|
cv2.namedWindow(f'{vid_name}', cv2.WINDOW_NORMAL | cv2.WINDOW_KEEPRATIO)
|
||||||
|
while True:
|
||||||
|
key = cv2.waitKeyEx(10)
|
||||||
|
|
||||||
|
# Close button(X button) event
|
||||||
|
if cv2.getWindowProperty(f'{vid_name}', cv2.WND_PROP_VISIBLE) < 1 :
|
||||||
|
return
|
||||||
|
|
||||||
|
if key > 0 :
|
||||||
|
# 13 == Enter key ASCII Code
|
||||||
|
if key == 13:
|
||||||
|
break
|
||||||
|
# Pause
|
||||||
|
elif key == ord('p') or key == ord('P'):
|
||||||
|
if isPaused :
|
||||||
|
isPaused = False
|
||||||
|
else :
|
||||||
|
isPaused = True
|
||||||
|
frame = copy.deepcopy(pausedFrame)
|
||||||
|
# Replay
|
||||||
|
elif key == ord('r') or key == ord('R'):
|
||||||
|
if not isPaused:
|
||||||
|
capture = cv2.VideoCapture(fpath)
|
||||||
|
continue
|
||||||
|
# Move Forward (10s)
|
||||||
|
elif key == ord('e') or key == ord('E'):
|
||||||
|
if not isPaused:
|
||||||
|
capture.set(cv2.CAP_PROP_POS_FRAMES, frame_pos + (fps * 10))
|
||||||
|
# Move Backward (10s)
|
||||||
|
elif key == ord('q') or key == ord('Q'):
|
||||||
|
if not isPaused:
|
||||||
|
capture.set(cv2.CAP_PROP_POS_FRAMES, frame_pos - (fps * 10))
|
||||||
|
elif key == ord('h') or key == ord('H'):
|
||||||
|
print("[INFO]\t ### KEY GUIDE ########################################################################################################")
|
||||||
|
print("[INFO]\t # (MOVE : wasd), (RESIZE : ijkl), (FOR / BACKWARD : q, e), (RESET : r), (PAUSE : p), (SKIP : n), (CONFIRM : enter) #")
|
||||||
|
print("[INFO]\t ###################################################################################################################### ")
|
||||||
|
# Skip this Vid
|
||||||
|
elif key == ord('n') or key == ord('N'):
|
||||||
|
break;
|
||||||
|
else:
|
||||||
|
# Crop Area control
|
||||||
|
cropPos = resizeCropInfo(key, cropPos, frame_width, frame_height)
|
||||||
|
|
||||||
|
frame_pos = capture.get(cv2.CAP_PROP_POS_FRAMES)
|
||||||
|
frame_count = capture.get(cv2.CAP_PROP_FRAME_COUNT)
|
||||||
|
|
||||||
|
if(frame_pos >= frame_count):
|
||||||
|
break
|
||||||
|
|
||||||
|
print("[INFO]\t Current crop area : ", end="")
|
||||||
|
print(cropPos, end="")
|
||||||
|
print(f"\t Frame Count ({frame_pos}/{frame_count})\t[PRESS \"h\" TO VIEW KEY GUIDE]", end="\r")
|
||||||
|
|
||||||
|
if not isPaused:
|
||||||
|
ret, frame = capture.read()
|
||||||
|
pausedFrame = copy.deepcopy(frame)
|
||||||
|
cv2.rectangle(frame, (cropPos[2], cropPos[3]), (cropPos[2] + cropPos[0],
|
||||||
|
cropPos[3] + cropPos[1]), (0, 0, 255), thickness=1, lineType=cv2.LINE_8)
|
||||||
|
cv2.imshow(f'{vid_name}', frame)
|
||||||
|
else :
|
||||||
|
pausedFrame = copy.deepcopy(frame)
|
||||||
|
cv2.rectangle(pausedFrame, (cropPos[2], cropPos[3]), (cropPos[2] + cropPos[0],
|
||||||
|
cropPos[3] + cropPos[1]), (0, 0, 255), thickness=1, lineType=cv2.LINE_8)
|
||||||
|
cv2.imshow(f'{vid_name}', pausedFrame)
|
||||||
|
|
||||||
|
print("")
|
||||||
|
capture.release()
|
||||||
|
cv2.destroyAllWindows()
|
||||||
|
|
||||||
|
# Debugging
|
||||||
|
# cropPos[0] = 678
|
||||||
|
# cropPos[1] = 402
|
||||||
|
# cropPos[2] = 20
|
||||||
|
# cropPos[3] = 38
|
||||||
|
# pad_arg = ""
|
||||||
|
|
||||||
|
dt_str = getTime()
|
||||||
|
|
||||||
|
if not os.path.exists("./logs"):
|
||||||
|
print(f"[INFO]\t Make log directory [./logs/{vid_name}.txt]")
|
||||||
|
os.makedirs("./logs")
|
||||||
|
|
||||||
|
if not os.path.exists(f"./logs/{vid_name}.txt"):
|
||||||
|
print(f"[INFO]\t Save log file in [./logs/{vid_name}.txt]")
|
||||||
|
f = open(f"./logs/{vid_name}.txt", 'w')
|
||||||
|
else :
|
||||||
|
print(f"[INFO]\t Add line in log file [./logs/{vid_name}.txt]")
|
||||||
|
f = open(f"./logs/{vid_name}.txt", 'a')
|
||||||
|
|
||||||
|
f.write(f"{dt_str}\tcrop={cropPos[0]}:{cropPos[1]}:{cropPos[2]}:{cropPos[3]}\n")
|
||||||
|
f.close
|
||||||
|
|
||||||
# ffmpeg filter args
|
# ffmpeg filter args
|
||||||
crop_arg = f"crop={cropPos[0]}:{cropPos[1]}:{cropPos[2]}:{cropPos[3]},"
|
crop_arg = f"crop={cropPos[0]}:{cropPos[1]}:{cropPos[2]}:{cropPos[3]},"
|
||||||
print(f"[INFO]\t [CropInfo] (W : {cropPos[0]}), (H : {cropPos[1]}), (X : {cropPos[2]}), (Y : {cropPos[3]})")
|
print(f"[INFO]\t [CropInfo] (W : {cropPos[0]}), (H : {cropPos[1]}), (X : {cropPos[2]}), (Y : {cropPos[3]})")
|
||||||
|
|
||||||
|
# 2021_07_28 : Delete padding argument
|
||||||
|
crop_arg = f"crop={cropPos[0]}:{cropPos[1]}:{cropPos[2]}:{cropPos[3]}"
|
||||||
crop_ratio = cropPos[0] / cropPos[1]
|
crop_ratio = cropPos[0] / cropPos[1]
|
||||||
scale_arg = ""
|
scale_arg = ""
|
||||||
|
pad_arg = ""
|
||||||
|
|
||||||
# Set padding(letter box) argument
|
# Set padding(letter box) argument
|
||||||
# variable "offset" for 16:9 output display ratio
|
# variable "offset" for 16:9 output display ratio
|
||||||
if target_ratio < crop_ratio:
|
# if target_ratio < crop_ratio:
|
||||||
offset = (9 * cropPos[0] - 16 * cropPos[1]) / 16
|
# offset = (9 * cropPos[0] - 16 * cropPos[1]) / 16
|
||||||
tmp_height = cropPos[1] + offset
|
# tmp_height = cropPos[1] + offset
|
||||||
tmp_height = int(tmp_height)
|
# tmp_height = int(tmp_height)
|
||||||
if (tmp_height % 4) != 0:
|
# if (tmp_height % 4) != 0:
|
||||||
tmp_height -= (tmp_height % 4)
|
# tmp_height -= (tmp_height % 4)
|
||||||
# scale_arg = f"scale=w={cropPos[0]}:h={tmp_height},"
|
# # scale_arg = f"scale=w={cropPos[0]}:h={tmp_height},"
|
||||||
pad_arg = f"pad={cropPos[0]}:{tmp_height}:(ow-iw)/2:(ih-oh)/2"
|
# pad_arg = f"pad={cropPos[0]}:{tmp_height}:(ow-iw)/2:(ih-oh)/2"
|
||||||
print(f"[INFO]\t [PadInfo] (W : {cropPos[0]}), (H : {tmp_height})")
|
# print(f"[INFO]\t [PadInfo] (W : {cropPos[0]}), (H : {tmp_height})")
|
||||||
else:
|
# else:
|
||||||
offset = (16 * cropPos[1] - 9 * cropPos[0]) / 9
|
# offset = (16 * cropPos[1] - 9 * cropPos[0]) / 9
|
||||||
tmp_width = cropPos[0] + offset
|
# tmp_width = cropPos[0] + offset
|
||||||
tmp_width = int(tmp_width)
|
# tmp_width = int(tmp_width)
|
||||||
if (tmp_width % 4) != 0:
|
# if (tmp_width % 4) != 0:
|
||||||
tmp_width -= (tmp_width % 4)
|
# tmp_width -= (tmp_width % 4)
|
||||||
# scale_arg = f"scale=w={tmp_width}:h={cropPos[1]},"
|
# # scale_arg = f"scale=w={tmp_width}:h={cropPos[1]},"
|
||||||
pad_arg = f"pad={tmp_width}:{cropPos[1]}:(ow-iw)/2:(ih-oh)/2"
|
# pad_arg = f"pad={tmp_width}:{cropPos[1]}:(ow-iw)/2:(ih-oh)/2"
|
||||||
print(f"[INFO]\t [PadInfo] (W : {tmp_width}), (H : {cropPos[1]})")
|
# print(f"[INFO]\t [PadInfo] (W : {tmp_width}), (H : {cropPos[1]})")
|
||||||
|
|
||||||
print("[INFO]\t Encoding \"%s\" ..." % vid_name)
|
print("[INFO]\t Encoding \"%s\" ..." % vid_name)
|
||||||
|
|
||||||
@@ -293,17 +351,34 @@ def main():
|
|||||||
|
|
||||||
output_path = os.path.join(output_dir, f"{vid_name}_box.mp4")
|
output_path = os.path.join(output_dir, f"{vid_name}_box.mp4")
|
||||||
if args.debug:
|
if args.debug:
|
||||||
# for Debug
|
# for Debug (short vid encoding)
|
||||||
p = subprocess.Popen(["ffmpeg", "-t", "15", "-ss", "30", "-i", fpath, "-vf", crop_arg + scale_arg + pad_arg, "-y",
|
if (i + 1) == vid_num :
|
||||||
"-pix_fmt", "yuv420p", "-c:v", "libx264", "-crf", "0", "-preset", "medium", "-loglevel", "error", f"{output_path}"])
|
subprocess.run(["ffmpeg", "-t", "15", "-ss", "30", "-i", fpath, "-vf", crop_arg + scale_arg + pad_arg, "-y",
|
||||||
|
"-pix_fmt", "yuv420p", "-c:v", "libx264", "-crf", "0", "-c:a", "mp3", "-b:a", "320k", "-preset", "medium", "-loglevel", "error", f"{output_path}"])
|
||||||
|
else :
|
||||||
|
p = subprocess.Popen(["ffmpeg", "-t", "15", "-ss", "30", "-i", fpath, "-vf", crop_arg + scale_arg + pad_arg, "-y",
|
||||||
|
"-pix_fmt", "yuv420p", "-c:v", "libx264", "-crf", "0", "-c:a", "mp3", "-b:a", "320k", "-preset", "medium", "-loglevel", "error", f"{output_path}"])
|
||||||
|
elif (logs is '') or manual_mode:
|
||||||
|
# Manual Mode
|
||||||
|
# Hold process when encoding last video
|
||||||
|
if (i + 1) == vid_num :
|
||||||
|
subprocess.run(["ffmpeg", "-i", fpath, "-vf", crop_arg + scale_arg + pad_arg, "-y", "-pix_fmt",
|
||||||
|
"yuv420p", "-c:v", "libx264", "-crf", "0", "-c:a", "mp3", "-b:a", "320k", "-preset", "medium", "-loglevel", "error", f"{output_path}"])
|
||||||
|
else :
|
||||||
|
p = subprocess.Popen(["ffmpeg", "-i", fpath, "-vf", crop_arg + scale_arg + pad_arg, "-y", "-pix_fmt",
|
||||||
|
"yuv420p", "-c:v", "libx264", "-crf", "0", "-c:a", "mp3", "-b:a", "320k", "-preset", "medium", "-loglevel", "error", f"{output_path}"])
|
||||||
else:
|
else:
|
||||||
p = subprocess.Popen(["ffmpeg", "-i", fpath, "-vf", crop_arg + scale_arg + pad_arg, "-y", "-pix_fmt",
|
# Log file mode (encoding simultaneously 5 videos)
|
||||||
"yuv420p", "-c:v", "libx264", "-crf", "0", "-preset", "medium", "-loglevel", "error", f"{output_path}"])
|
if (i % 4 == 0) and (i != 0):
|
||||||
|
subprocess.run(["ffmpeg", "-i", fpath, "-vf", crop_arg + scale_arg + pad_arg, "-y", "-pix_fmt",
|
||||||
# print("Encoding Done.")
|
"yuv420p", "-c:v", "libx264", "-crf", "0", "-c:a", "mp3", "-b:a", "320k", "-preset", "medium", "-loglevel", "error", f"{output_path}"])
|
||||||
|
else :
|
||||||
|
p = subprocess.Popen(["ffmpeg", "-i", fpath, "-vf", crop_arg + scale_arg + pad_arg, "-y", "-pix_fmt",
|
||||||
|
"yuv420p", "-c:v", "libx264", "-crf", "0", "-c:a", "mp3", "-b:a", "320k", "-preset", "medium", "-loglevel", "error", f"{output_path}"])
|
||||||
|
|
||||||
# debugging
|
# debugging
|
||||||
# return
|
# return
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
main()
|
main()
|
||||||
|
print("[INFO]\t Encoding Done.")
|
||||||
Reference in new issue
Block a user