YOLOv1 理论与实现
前言
本人使用的tensorflow-v2 遇到许多问题,均已经成功解决。
所有问题与解决均已经分享在本文。
版本要求 | 版本要求 |
---|---|
Tensorflow >= 1.0 | OpenCV_Python >= 4.2 |
一、配置流程
-
下载程序包
$ git clone https://github.com/hizhangp/yolo_tensorflow.git $ cd yolo_tensorflow
-
下载 pascal 数据集
$ ./download_data.sh
失败的请手动下载: **网盘. 提取码:aiAI
- 下载权重
方式1==》谷歌下载.
方式2==》
YOLO_small.提取码:aiAI
并且 放在 data/weight
目录下
-
修改参数配置
yolo/config.py
-
Training
$ python train.py
-
Test
$ python test.py
二、问题与解答
-
Tensorflow bug(一) ValueError The passed save_path is not a valid checkpoint
原因:路径的相对引用失败
图片: -
AttributeError: module ‘tensorflow.contrib’ has no attribute ‘slim’
原因:安装的是tensorflow v2版本的问题
解决:
-
tf.placeholder() is not compatible with eager execution
原因:安装的是tensorflow v2版本的问题
解决:
-
AttributeError: ‘NoneType’ object has no attribute ‘shape’
原因:在使用 cv2.imread(imname) 传入图片时,imname的路径没找到,即传入图片失败
解决:
最后得到的效果:基于cpu 跑出来的效果
参考部分
YOLO v1理论参考:
链接: 参考.
源码下载链接:
链接: GitHub.