@ptrblck thank you. I did that but now does have 2 channels RuntimeError: Given groups=1, weight of size [5, 1, 3, 3], expected input[1, 2, 240, 320] to have 1 channels, but got 2 channels instead
t_image = image.convert('LA') suppose to convert to grayscal image? Is that right?