sunshineatnoon / Darknet.keras

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Darknet.keras

This is a transfer from weights trained by Darknet to keras with Theano backend.

Currently I only finish test process for YOLO Tiny Model.

Due to image preprocess difference bewteen my reimplementaion and Darknet's, the result has slightly difference.

To run the code:

  1. Create three folders: weights, results, images.
  2. Put the images you want to detect in the folder images.
  3. Download yolo-tiny.weights from Darknet website and put it in the folder weights.
  4. Run:
    python RunTinyYolo.py
    

The detection result will be saved in the results folder

Notes: Make sure your have Theano and Keras installed

IMPORTANT WARNING

This code is written for older version of YOLO weight, now the weight has updated to version 2.0, so the code is no longer working. In order to make it work, one has to change ReadTinyYOLONetWeights to read new YOLO weight bite by bite.

About


Languages

Language:Python 100.0%