The issue of change your code?

[Jayhello]

I want to change you code to train rgb image? In the below image, I guess the 1 means gray image? But the image to train, it's width and height does not indicate ? If I change the image to rgb , I change " local nIn = nm[i-1] or 1 => local nIn = nm[i-1] or 3 " is that?