module python files
This commit is contained in:
JunHo Kim committed
1 parent
ff088e051d
commit
67fdffbe10
5 files changed
+531
-490
No files matched your search
@@ -0,0 +1,212 @@
|
||||
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
|
||||
|
||||
Reference in new issue
Block a user