ValueError: Ошибка при проверке ввода: ожидалось, что input_1 будет иметь 4 измерения, но получил массив с формой (6243, 256, 256)

Ваш входной слой ожидает массив формы (batch_size, 256, 256, 1), но похоже, что вы передаете данные формы (batch_size, 256, 256). Вы можете попробовать изменить свои тренировочные данные следующим образом:

tr_img_data = np.expand_dims(tr_img_data, axis=-1)