Expected shape (1, 400, 600, 6) but found (1, 400, 600, 4)
Expected shape (1, 400, 600, 6) but found (1, 400, 600, 4)