bhaktatejas922:

I narrowed it down to it being albementations that converts it to 3 channels ‘sometimes’. Still can’t pinpoint why its doing it.


I would then try to use the already mentioned approach of isolating the Dataset index, which is running into the assert and check how albumentations is transforming this particular sample.