Update train.py

e5aaf662 · Bubbliiiing · GitHub · da37b95e · e5aaf662
显示空白变更内容
内联并排

Showing with 11 addition and 10 deletion

train.py train.py +11 -10

未找到文件。
--- a/train.py
+++ b/train.py
@@ -106,6 +106,7 @@ if __name__ == "__main__":
    #------------------------------------------------------#
    model = YoloBody(anchors_mask, num_classes)
    weights_init(model)
+    if model_path != '':
        #------------------------------------------------------#
        #   权值文件请看README，百度网盘下载
        #------------------------------------------------------#