From 3a9796908b6198804071d3e71a8959a13526ca5e Mon Sep 17 00:00:00 2001
From: Bubbliiiing <47347516+bubbliiiing@users.noreply.github.com>
Date: Thu, 11 Nov 2021 23:57:13 +0800
Subject: [PATCH] Update unet.py

---
 unet.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/unet.py b/unet.py
index c03e023..12157e2 100644
--- a/unet.py
+++ b/unet.py
@@ -23,7 +23,7 @@ class Unet(object):
         #   训练好后logs文件夹下存在多个权值文件，选择验证集损失较低的即可。
         #   验证集损失较低不代表miou较高，仅代表该权值在验证集上泛化性能较好。
         #-------------------------------------------------------------------#
-        "model_path"        : 'model_data/unet_voc',
+        "model_path"        : 'model_data/unet_voc.h5',
         #----------------------------------------#
         #   所需要区分的类的个数+1
         #----------------------------------------#