前言
本文记录我在学习YOLO-v1过程中参考的相关资料以及部分个人理解。
原理
相关理论知识,网络上有许多参考资料,我主要通过这篇博客来学习。
tensorflow实现
我实现的代码都有详细注释,这里不再解释
相关参考
参考blog:
xiaohu版本代码
hizhangp版本代码
参考github:
hizhangp
如何进行分类模型的预训练:
blog
数据准备
下载weight文件YOLO_small
下载voc_07数据集,网上都可以找到
训练
执行train.py文件即可。可以从0开始训练,也可以从以前模型进行训练。