12345678910111213141516171819202122232425262728293031323334353637383940 |
- 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
- 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(".avi")):
- 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)
|