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