YOLOv1 理论与实现

YOLOv1 理论与实现

前言
本人使用的tensorflow-v2 遇到许多问题，均已经成功解决。
所有问题与解决均已经分享在本文。

版本要求	版本要求
Tensorflow >= 1.0	OpenCV_Python >= 4.2

一、配置流程

1. 下载程序包

   $ git clone https://github.com/hizhangp/yolo_tensorflow.git
   $ cd yolo_tensorflow
   

2. 下载 pascal 数据集

   $ ./download_data.sh
   

失败的请手动下载： **网盘. 提取码：aiAI

3. 下载权重

方式1==》谷歌下载.
方式2==》
YOLO_small.提取码：aiAI
并且 放在 data/weight 目录下

4. 修改参数配置 yolo/config.py

5. Training

   $ python train.py
   

6. Test

   $ python test.py
   

二、问题与解答

1. Tensorflow bug(一) ValueError The passed save_path is not a valid checkpoint
   原因：路径的相对引用失败
   图片:

2. AttributeError: module ‘tensorflow.contrib’ has no attribute ‘slim’
   原因：安装的是tensorflow v2版本的问题
   解决：
   

3. tf.placeholder() is not compatible with eager execution
   原因：安装的是tensorflow v2版本的问题
   解决：
   

4. AttributeError: ‘NoneType’ object has no attribute ‘shape’
   原因：在使用 cv2.imread(imname) 传入图片时，imname的路径没找到，即传入图片失败
   解决：

最后得到的效果：基于cpu 跑出来的效果

参考部分

YOLO v1理论参考：
链接: 参考.

源码下载链接：
链接: GitHub.