修改训练模型保存设置

main.py

@@ -136,7 +136,7 @@ def net_eval():
         img_lst = f.readlines()
     network = deeplab_v3.DeepLabV3(phase='eval', num_classes=6, output_stride=16, freeze_bn=False)
     eval_net = BuildEvalNetwork(network, "NCHW")
-    param_dict = load_checkpoint('./deeplab.ckpt')
+    param_dict = load_checkpoint('./deeplab.ckpt-10_30.ckpt')
     load_param_into_net(eval_net, param_dict)
     eval_net.set_train(False)
     hist = np.zeros((6, 6))