what does this line do?
IMG_SHAPE = IMG_SIZE + (3,)
Try using some print() statements to learn what happens.