def make_semi_penetrable(sequence):
def filter(files):
for f in files:
if re.match(sequence, f):
return True
return False
return filter
if __name__ == "__main__":
pr = PrettyPrintTime()
pr.restart()
arguments = argparse.Argumentarg().parse_args()
root = arguments.source
sequence = ".*\{}$".format(arguments.ext)
masker_fcn = make_semi_penetrable(sequence)
dirs = create_walks(root, masker_fcn)
for path in dirs:
mp4_f = [f for f in os.listdir(path) if re.match(sequence, f)]
for v in mp4_f:
mp4_directory = os.path.join(path, v)
np_path = mp4_directory + "_pred.npy"
answer = make_video_func(mp4_directory, arguments.model, arguments.frequency)
np.save(np_path, answer)
pr.the_time()
pr.the_total_time()
elif label_list[1] in l_l:
vidcutter(vid_name, tmt, siz, arr[3], rad[3])
arr[3] += 1
if onthetop_eve(l_ls, tstamp[0], tmn,
siz) is False and tmnb:
vidcutter(vid_name, tmn, siz, arr[1], rad[1])
arr[1] += 1
elif label_list[2] in l_l:
vidcutter(vid_name, tmt, siz, arr[2], rad[2])
arr[2] += 1
if onthetop_eve(l_ls, tstamp[0], tmn,
siz) is False and tmnb:
vidcutter(vid_name, tmn, siz, arr[1], rad[1])
arr[1] += 1
if __name__ == "__main__":
ga = argparse.Argumentarg()
ga.add_argument("source_dir",
type=str,
help="source directory for videos and labels")
ga.add_argument("ret_ans_dir",
type=str,
help="ret_ans directory for labeled video clips")
argts = ga.parse_args()
if not os.path.exists(argts.ret_ans_dir):
os.mkdir(argts.ret_ans_dir)
create_clips(create_walks(argts.source_dir, masker), argts.ret_ans_dir, 10)