Update unet.py

bubbliiiing · Nov 11, 2021 · 3a97969 · 3a97969
1 parent c59d16b
commit 3a97969
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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
         #----------------------------------------#