| 160 | |
| 161 | |
| 162 | class KITTI(FlowDataset): |
| 163 | def __init__(self, aug_params=None, split='training', root='datasets/KITTI'): |
| 164 | super(KITTI, self).__init__(aug_params, sparse=True) |
| 165 | if split == 'testing': |
| 166 | self.is_test = True |
| 167 | |
| 168 | root = osp.join(root, split) |
| 169 | images1 = sorted(glob(osp.join(root, 'image_2/*_10.png'))) |
| 170 | images2 = sorted(glob(osp.join(root, 'image_2/*_11.png'))) |
| 171 | |
| 172 | for img1, img2 in zip(images1, images2): |
| 173 | frame_id = img1.split('/')[-1] |
| 174 | self.extra_info += [ [frame_id] ] |
| 175 | self.image_list += [ [img1, img2] ] |
| 176 | |
| 177 | if split == 'training': |
| 178 | self.flow_list = sorted(glob(osp.join(root, 'flow_occ/*_10.png'))) |
| 179 | |
| 180 | |
| 181 | class HD1K(FlowDataset): |