在本节课程中,需要用到YOLO官方预训练的模型,官方的模型包括两个文件cfg(网络配置)和weight(参数权重),用到keras中的话需要转换成h5文件。因此需要利用YAD2K(Yet Another Darknet 2 Keras https://github.com/allanzelener/YAD2K)项目,完成这个转换。
YAD2K项目中Readme文件有使用方法,Linux下懒人可以直接按照顺序操作完成转换,Windows下进行稍微改变,也可以完成转换。https://blog.csdn.net/overfit/article/details/78744947这篇博客作者写了自己的做法。
最关键的一点,yolo.cfg和yolo.weights两个文件有无数大坑,需要下载合适的文件才能完成转换。yolo官网下载速度很慢,可以从此链接下载https://download.csdn.net/download/zbwindboy/10294400。如果完全不想自己尝试,那么可以直接下载转换好的模型https://download.csdn.net/download/zbwindboy/10294431。