(sintel_dir)
| 163 | cv2.imwrite(viz_fn, flow_img[:, :, [2,1,0]]) |
| 164 | |
| 165 | def process_sintel(sintel_dir): |
| 166 | img_pairs = [] |
| 167 | for scene in os.listdir(sintel_dir): |
| 168 | dirname = osp.join(sintel_dir, scene) |
| 169 | image_list = sorted(glob(osp.join(dirname, '*.png'))) |
| 170 | for i in range(len(image_list)-1): |
| 171 | img_pairs.append((image_list[i], image_list[i+1])) |
| 172 | |
| 173 | return img_pairs |
| 174 | |
| 175 | def generate_pairs(dirname, start_idx, end_idx): |
| 176 | img_pairs = [] |