DocumentAIKit/Dewarp/smartdoc_data_processor/video_to_image.py

import os


def argsProcessor():
    import argparse
    parser = argparse.ArgumentParser()
    parser.add_argument("-d", "--dataPath", help="path to main data folder")
    parser.add_argument("-o", "--outputPath", help="output data")
    return parser.parse_args()


if __name__ == '__main__':
    args = argsProcessor()
    dir = args.dataPath
    output = args.outputPath
    print(dir)
    if not os.path.isdir(output):
        os.mkdir(output)

    for folder in os.listdir(dir):
        if os.path.isdir(dir + "/" + folder):
            dir_temp = dir + '/' + folder + "/"
            for file in os.listdir(dir_temp):
                print(file)
                from subprocess import call

                if file.endswith(".mp4"):
                    call("mkdir " + output + folder, shell=True)
                    if os.path.isdir(output + folder + "/" + file):
                        print("Folder already exist")
                    else:
                        call("cd " + output + folder + " && mkdir " + file, shell=True)
                        call("ls", shell=True)

                        location = dir + folder + "/" + file
                        gt_address = "cp " + location[
                                             0:-4] + ".gt.xml " + output + folder + "/" + file + "/" + file + ".gt"
                        call(gt_address, shell=True)
                        command = "ffmpeg -i " + location + " " + output + folder + "/" + file + "/%3d.jpg"
                        print(command)
                        call(command, shell=True)