Всё починил. Дело в том, как я подавал входные данные. Я делал это, подавая весь массив, и Keras считал это как множество входных данных (а именно 244 массива) размером 244 на 3. Нужно добавить строчки:
m_data = np.ndarray(shape=(1, 244, 244, 3), dtype=np.float32)
m_data[0] = normalized_image_array
И тогда всё будет нормально.