From 6a688ba29ae6f429dec68a22f697d6c275a57975 Mon Sep 17 00:00:00 2001
From: Bubbliiiing <47347516+bubbliiiing@users.noreply.github.com>
Date: Thu, 11 Jun 2020 11:41:13 +0800
Subject: [PATCH] Update train.py

---
 train.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/train.py b/train.py
index 539a36d..d78da49 100644
--- a/train.py
+++ b/train.py
@@ -142,7 +142,10 @@ if __name__ == "__main__":
     # 获取classes和anchor的位置
     classes_path = 'model_data/voc_classes.txt'    
     anchors_path = 'model_data/yolo_anchors.txt'
-    # 预训练模型的位置
+    #-------------------------------------------#
+    #   权值文件的下载请看README
+    #   预训练模型的位置
+    #-------------------------------------------#
     weights_path = 'model_data/yolo4_weight.h5'
     # 获得classes和anchor
     class_names = get_classes(classes_path)
@@ -173,9 +176,6 @@ if __name__ == "__main__":
     
     # 载入预训练权重
     print('Load weights {}.'.format(weights_path))
-    #-------------------------------------------#
-    #   权值文件的下载请看README
-    #-------------------------------------------#
     model_body.load_weights(weights_path, by_name=True, skip_mismatch=True)
     
     # y_true为13,13,3,85
-- 
GitLab