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模型。