@ptrblck, Yeah. I am testing it right now. I have one more question, if I convert my input to RGB and use, can I go ahead with any batch_size. Right now, I am using batch_size = 1.
But, in some posts, I found the batch_size also creates problem.