Проблема с построением прогнозируемых масок ⇐ Python

[Anonymous]

В настоящее время я работаю над проектом глубокого обучения «Сегментация болезней листьев». Я обучил модель за 50 эпох и получил следующие показатели точности и потерь:
Потери поезда: 19,4736, Точность поезда: 0,9395
Потери при проверке: 19,6197, Точность проверки: 0,9100
Потери при тестировании: 19,6148, Точность теста: 0,9123
Однако, когда я рисую предсказанные маски, они кажутся неточными. Есть ли проблемы с моим кодом построения графиков?

Код: Выделить всё

def plot_predictions(model, images, masks, num_samples=5):
predictions = model.predict(images[:num_samples])
for i in range(num_samples):
plt.figure(figsize=(15, 5))
plt.subplot(1, 3, 1)
plt.title('Real Image')
plt.imshow(images[i])
plt.subplot(1, 3, 2)
plt.title('Ground Truth Mask')
plt.imshow(masks[i], cmap='gray')  # Assuming masks are already binary
plt.subplot(1, 3, 3)
plt.title('Predicted Mask')
plt.imshow(predictions[i][:, :, 0], cmap='gray')  # Convert predicted mask to binary
plt.show()

plot_predictions(model, test_images.numpy(), test_masks_L, num_samples=5)

Исходное изображение-маска-предсказанная маска
Проверьте мой код и помогите выявить ошибки, которые могут вызвать эту проблему?

Подробнее здесь: https://stackoverflow.com/questions/786 ... cted-masks