Failed Algorithm

Hello, would it be possible to get feedback on why my algorithm (UNet + Macenko) failed please? I have verified that it works on a full 2mm image from the training set so this is a mystery to me

Here you go:

2021-08-26T18:38:54.444979864Z File "/opt/algorithm/detection.py", line 81, in process_image 2021-08-26T18:38:54.444987034Z last_dim=input_image[:,:,3] 2021-08-26T18:38:54.444993827Z IndexError: index 3 is out of bounds for axis 2 with size 3