/keras_yolov4

Primary LanguagePythonMIT LicenseMIT

keras-yolo4

license

Introduction

A Keras implementation of YOLOv4 (Tensorflow backend) inspired by https://github.com/AlexeyAB/darknet.

Frame code from https://github.com/qqwweee/keras-yolo3.

Quick Start

  1. Download YOLOv4 weights from yolov4.weights.
  2. Convert the Darknet YOLOv4 model to a Keras model.
  3. Run YOLOv4 detection.
python convert.py

Running convert.py will get keras yolov4 weight file yolo4_weight.h5.

Usage

Inference

python test.py

Todo

Debug train code with CIoU loss. Loss code from https://github.com/david8862/keras-YOLOv3-model-set.