YAD2K使用介绍

YAD2K是Yet Another Darknet 2 Keras的缩写。

这是由90%的Keras和10%的Tensorflow实现的YOLO_v2.

依赖：

?Keras

?Tensorflow

?Numpy

?h5py (For Keras model serialization.)

?Pillow (For rendering test results.)

?Python 3

?pydot-ng (Optional for plotting model.)

安装：

1、下载yad2k

git clone https://github.com/allanzelener/yad2k.git

2、安装依赖包

pip install numpy h5py pillow

pip install tensorflow-gpu # CPU-only: conda install -c conda-forge tensorflow

pip install keras # Possibly older release: conda install keras

开始：

1、下载Darknet模型 cfg，权重。

2、转换YOLO v2模型为keras模型

./yad2k.py yolo.cfg yolo.weights model_data/yolo.h5

3、用images/下的测试集，测试转换后的模型。

./test_yolo.py model_data/yolo.h5

其他说明：

voc_conversion_scripts包含了2个转换带有XML标注的Pascal VOC数据集为HDF5或TFRecords格式的脚本

yad2k/models包含了Darknet-19与YOLO_v2的实现。

train_overfit是一个样例的训练脚本，它过拟合了来自Pascal VOC数据集的一个单独图像的YOLO v2模型。