213 lines
6.2 KiB
Python
213 lines
6.2 KiB
Python
import os
|
|
from collections import Counter
|
|
import cv2
|
|
import glob
|
|
import re
|
|
import subprocess
|
|
import argparse
|
|
import copy
|
|
import datetime
|
|
import time
|
|
|
|
def resizeCropInfo(key, cropPos, frame_width, frame_height):
|
|
# Adjust crop info in display layout.
|
|
def _IsOutOfBound(_cropPos):
|
|
w, h, x, y = _cropPos
|
|
if (w + x) > frame_width or (h + y) > frame_height:
|
|
return True
|
|
if (x < 0) or (y < 0):
|
|
return True
|
|
if (w <= 0) or (h <= 0):
|
|
return True
|
|
|
|
w, h, x, y = cropPos
|
|
# Left (Expand width)
|
|
if key == ord('j') or key == ord('J'):
|
|
x -= 2
|
|
w += 4
|
|
# Right (Shrink width)
|
|
elif key == ord('l') or key == ord('L'):
|
|
x += 2
|
|
w -= 4
|
|
# Up (Expand height)
|
|
elif key == ord('i') or key == ord('I'):
|
|
y -= 2
|
|
h += 4
|
|
# Down (Shrink height)
|
|
elif key == ord('k') or key == ord('K'):
|
|
y += 2
|
|
h -= 4
|
|
# Num 4 (Move left)
|
|
elif key == ord('a') or key == ord('A'):
|
|
# w -= 1
|
|
x -= 1
|
|
# Num 6 (Move right)
|
|
elif key == ord('d') or key == ord('D'):
|
|
# w += 1
|
|
x += 1
|
|
# Num 8 (Move upward)
|
|
elif key == ord('w') or key == ord('W'):
|
|
# h -= 1
|
|
y -= 1
|
|
# Num 2 (Move downward)
|
|
elif key == ord('s') or key == ord('S'):
|
|
# h += 1
|
|
y += 1
|
|
_cropPos = w, h, x, y
|
|
|
|
if not _IsOutOfBound(_cropPos):
|
|
cropPos = _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 getVideDuration(file_path):
|
|
ffprobe_command = f"ffprobe -v error -show_entries format=duration -of default=noprint_wrappers=1:nokey=1 \"{file_path}\""
|
|
p = subprocess.run(ffprobe_command, shell=False, stdout=subprocess.PIPE)
|
|
_hour = 0
|
|
_min = 0
|
|
_sec = 0
|
|
|
|
_sec = int(float(p.stdout.decode('utf-8')))
|
|
# print(_sec)
|
|
|
|
if _sec > 60 :
|
|
_min = int(_sec // 60)
|
|
_sec = _sec - (60 * _min)
|
|
# print(_min, _sec)
|
|
if _sec > 3600 :
|
|
_hour = int(_sec // 60)
|
|
_sec = _sec - (60 * _min)
|
|
_min = int(_sec // 60)
|
|
_sec = _sec - (60 * _min)
|
|
# print(_hour, _min, _sec)
|
|
|
|
return [_hour, _min, _sec]
|
|
|
|
# for unicode free filename
|
|
def imwrite(filename, img, params=None):
|
|
try:
|
|
ext = os.path.splitext(filename)[1]
|
|
result, n = cv2.imencode(ext, img, params)
|
|
if result:
|
|
with open(filename, mode='w+b') as f:
|
|
n.tofile(f)
|
|
return True
|
|
else:
|
|
return False
|
|
except Exception as e:
|
|
print(e)
|
|
return False
|
|
|
|
def parseLog(log_path, edit_mode, manual_mode, scale):
|
|
cropPos = []
|
|
|
|
if not os.path.exists(log_path):
|
|
print(f"[INFO]\t Can`t find log file. [{log_path}]")
|
|
manual_mode = True
|
|
edit_mode = False
|
|
else :
|
|
print(f"[INFO]\t Loading log file... [{log_path}]")
|
|
try :
|
|
f = open(log_path, "r")
|
|
logs = f.read()
|
|
|
|
# if log file is not empty
|
|
if logs != '':
|
|
# Delete "crop=" in list "crop"
|
|
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 * scale) for x in cropPos]
|
|
if scale > 1 :
|
|
print(f"[INFO]\t Scale up crop area: [w : {cropPos[0]}, h : {cropPos[1]}, x : {cropPos[2]}, y : {cropPos[3]}]")
|
|
|
|
else :
|
|
print(f"[!!! WARNNIG !!!]\t Empty log file. [{log_path}]")
|
|
manual_mode = True
|
|
edit_mode = False
|
|
except Exception as e:
|
|
print(f"[!!! WARNNIG !!!]\t Can`t read log file. [{log_path}]")
|
|
print(e)
|
|
manual_mode = True
|
|
edit_mode = False
|
|
|
|
|
|
return cropPos, edit_mode, manual_mode
|
|
|
|
def writeLog(logs_dir, log_path, cropPos):
|
|
if not os.path.exists(f"{logs_dir}"):
|
|
print(f"[INFO]\t Make log directory [{log_path}]")
|
|
os.makedirs(f"{logs_dir}")
|
|
|
|
if not os.path.exists(f"{log_path}"):
|
|
print(f"[INFO]\t Save log file in [{log_path}]")
|
|
f = open(f"{log_path}", 'w')
|
|
else :
|
|
print(f"[INFO]\t Add line in log file [{log_path}]")
|
|
f = open(f"{log_path}", 'a')
|
|
|
|
dt_str = getTime()
|
|
|
|
f.write(f"{dt_str}\tcrop={cropPos[0]}:{cropPos[1]}:{cropPos[2]}:{cropPos[3]}\n")
|
|
f.close
|
|
|
|
def autoDetectCropArea(fpath):
|
|
cropPos = []
|
|
|
|
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')
|
|
allCrops = re.findall("crop=\S+", infos)
|
|
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})")
|
|
return cropPos
|
|
|
|
# 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]
|
|
|
|
return cropPos
|
|
|
|
def adjustCropArea(cropPos):
|
|
# 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
|
|
|
|
return cropPos
|
|
|