未验证 提交 a6906ebc 编写于 作者: B Bubbliiiing 提交者: GitHub

Update train.py

上级 eac21b19
......@@ -173,6 +173,9 @@ 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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册