Hi,
while doing the conversion i’m getting length exception I posted the issue there:

  
    
    
    Getting the error while converting the tensors to Images using cv2 
  
  
    Hi, 
I have converted the tensors which I’m getting as an output(test set) from my UNet to images. When I’m testing the model it is throwing out of index exception which I’m not getting when I didn’t convert the tensors to images. I have 74 images in my test set but after getting tested with one image it is throwing out of index exception in my custom dataset class. any idea will there will be any problem if I do a conversion like that? 
# this is where im converting
 def test_step(self, batch, …
  


Please give any suggestion about this