Juartaurus/poi_Engineering

1import ffmpeg2import cv23import os4 5ALLOWED_EXTENSIONS = set(['txt', 'pdf', 'png', 'jpg', 'jpeg', 'gif'])6 7 8def allowed_file(filename):9    return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS10 11 12def check_rotation(path_video_file):13    meta_dict = ffmpeg.probe(path_video_file)14    rotateCode = None15    print(int(meta_dict['streams'][0]['tags']['rotate']))16    if int(meta_dict['streams'][0]['tags']['rotate']) == 90:17        rotateCode = cv2.ROTATE_18018    elif int(meta_dict['streams'][0]['tags']['rotate']) == 180:19        rotateCode = cv2.ROTATE_90_COUNTERCLOCKWISE20    elif int(meta_dict['streams'][0]['tags']['rotate']) == 270:21        rotateCode = cv2.ROTATE_90_CLOCKWISE22    return rotateCode23 24 25def correct_rotation(frame, rotateCode):26    return cv2.rotate(frame, rotateCode)27 28 29def create_folder(dir_path):30    if not os.path.isdir(dir_path):31        os.mkdir(dir_path)32    subfolder = ['frames', 'ss', 'sts', 'ir']33    subfolder = [os.path.join(dir_path, subf) for subf in subfolder]34    for subf in subfolder:35        if not os.path.isdir(subf):36            os.mkdir(subf)37    return subfolder38