Thanks for sharing! the trainning code just for color image ,are there any instructions to use your code for training gray image?