What is the use of 1 in here? Also what happens when i use 2 instead of 1?
np.array(X).reshape(-1, IMG_SIZE, IMG_SIZE, 1)
