(rgb, gt, modal_x)
| 14 | |
| 15 | |
| 16 | def random_mirror(rgb, gt, modal_x): |
| 17 | if random.random() >= 0.5: |
| 18 | rgb = cv2.flip(rgb, 1) |
| 19 | gt = cv2.flip(gt, 1) |
| 20 | modal_x = cv2.flip(modal_x, 1) |
| 21 | |
| 22 | return rgb, gt, modal_x |
| 23 | |
| 24 | |
| 25 | def random_scale(rgb, gt, modal_x, scales): |