x.shape:(1024, 2048, 3) y.shape:(1024, 2048)

100%|██████████| 20/20 [00:13<00:00, 1.54it/s]