Блог

При загрузке модели и использовании ее для вывода я загружаю массив изображений, image_tile_tensor где находится форма (total_tile, tile_height, tile_width, 3) . image_tile_tensor это массив numpy.

Я ввожу это в свою модель для вывода, используя приведенный ниже код:

 image_tile_tensor = tf.image.convert_image_dtype(image_tile_tensor, tf.float32)
image_tile_prediction = model.predict(image_tile_tensor, verbose=1)
  

Во время вывода все предсказания выполняются хорошо, за исключением нескольких последних фрагментов изображения в image_tile_tensor . Например, если у меня всего 112 фрагментов изображения для вывода, последние 12 фрагментов изображения имеют прогнозирование со всеми значениями, равными NaN , в то время как первые 100 фрагментов изображения имеют ожидаемое значение прогноза.

Есть идеи, в чем может быть проблема? Я немного потерялся и не знаю, с чего начать отладку этого в данный момент. Если это помогает, входные данные tile_height и tile_width являются (192,192) и обучающими batch size = 16 .