| 248 | |
| 249 | |
| 250 | def process_sintel(args, img_path): |
| 251 | if args.full_seq: |
| 252 | for dir in tqdm(sorted(glob.glob(f"{img_path}/*/"))): |
| 253 | filelist = sorted(glob.glob(f"{dir}/*.png")) |
| 254 | save_dir = f"{args.output_dir}/{os.path.basename(os.path.dirname(dir))}" |
| 255 | yield filelist, save_dir |
| 256 | else: |
| 257 | seq_list = [ |
| 258 | "alley_2", |
| 259 | "ambush_4", |
| 260 | "ambush_5", |
| 261 | "ambush_6", |
| 262 | "cave_2", |
| 263 | "cave_4", |
| 264 | "market_2", |
| 265 | "market_5", |
| 266 | "market_6", |
| 267 | "shaman_3", |
| 268 | "sleeping_1", |
| 269 | "sleeping_2", |
| 270 | "temple_2", |
| 271 | "temple_3", |
| 272 | ] |
| 273 | for seq in tqdm(seq_list): |
| 274 | filelist = sorted(glob.glob(f"{img_path}/{seq}/*.png")) |
| 275 | save_dir = f"{args.output_dir}/{seq}" |
| 276 | yield filelist, save_dir |