yolov2

yolo v2 is an object detection algorithm using deep learning.

This repository contains learning and testing in keras and tensorflow.

ToDo

• refactor trainning code

• tracking function(SORT and Deep SORT) based object detection of yolov2

• demo gif

Prerequirement

• python3.6

• opencv for python2 involving movie function

• dataset(image and annotation using XML format)

• keras(tensorflow)

Usage

train

If you want to learn with your dataset you need to edit config.json.

Please prepare the annotation in XML format with reference to yolov 2 formula.

python train.py -c config.json

test

python predict.py -c config.json -w path/to/pretrained_weights -i path/to/image -o path/to/output_dir

tracking

todo

Tracking code performed by SORT algorithm using Kalman filter.

DEMO

todo