YAD2K

> 编程语言

开源

YAD2K：又一个Darknet转Keras工具

2.7K stars0 点赞1 次浏览

访问官网 GitHub

工具介绍

YAD2K：又一个Darknet转Keras工具

YAD2K：又一个 Darknet 2 Keras 实现

欢迎使用 YAD2K

你只需看一次，但你会一遍又一遍地重新实现神经网络。

YAD2K 是 YOLO_v2 的一个 90% Keras / 10% Tensorflow 实现。

原始论文：YOLO9000：更好、更快、更强，作者为 Joseph Redmond 和 Ali Farhadi。

环境要求

Keras
Tensorflow
Numpy
h5py（用于 Keras 模型序列化。）
Pillow（用于渲染测试结果。）
Python 3
pydot-ng（可选，用于绘制模型。）

安装

bash

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

# [Option 1] To replicate the conda environment:
conda env create -f environment.yml
source activate yad2k
# [Option 2] Install everything globaly.
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

快速开始

从官方YOLO网站下载Darknet模型配置文件和权重。
将Darknet YOLO_v2模型转换为Keras模型。
在images/中的小型测试集上测试转换后的模型。

bash

wget http://pjreddie.com/media/files/yolo.weights
wget https://raw.githubusercontent.com/pjreddie/darknet/master/cfg/yolo.cfg
./yad2k.py yolo.cfg yolo.weights model_data/yolo.h5
./test_yolo.py model_data/yolo.h5  # output in images/out/

更多选项请参阅./yad2k.py --help和./test_yolo.py --help。