Some questions about code

[southbanyan]

Thanks a lot for sharing the code and it's really useful for me.
I have two questions about the code.

1. In function cvtColor (if len(np.shape(image)) == 3 and np.shape(image)[-2] == 3) should we change np.shape(image)[-2] == 3 to np.shape(image)[-1] == 3? I don't quite understand the meaning here.
2. I have run code with my own dataset and it has pretty good performance, however there was an overfitting. Is Earlystop a good way to avoid overfitting?

Looking forward to your answer. Thanks