(dirname, start_idx, end_idx)
| 173 | return img_pairs |
| 174 | |
| 175 | def generate_pairs(dirname, start_idx, end_idx): |
| 176 | img_pairs = [] |
| 177 | for idx in range(start_idx, end_idx): |
| 178 | img1 = osp.join(dirname, f'{idx:06}.png') |
| 179 | img2 = osp.join(dirname, f'{idx+1:06}.png') |
| 180 | # img1 = f'{idx:06}.png' |
| 181 | # img2 = f'{idx+1:06}.png' |
| 182 | img_pairs.append((img1, img2)) |
| 183 | |
| 184 | return img_pairs |
| 185 | |
| 186 | if __name__ == '__main__': |
| 187 | parser = argparse.ArgumentParser() |