lArkl / PedestrainDetection

Pedestrian Detection using Caltech and INRIA dataset with Darknet Framework.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

PedestrainDetection

Pedestrian Detection using INRIA dataset and YOLO model with Darknet framework.

I trained the model following the steps from AlexeyAB

Scripts

You can find the scripts that convert the annotations of the INRIA dataset into the labels required for Darknet.

You can find the converted labels in the INRIAPerson folder within the Train and Test folders respectively.

Also, you can graph your current AVG Loss with the ShowTraining script.

Training

The files inria.data and inria.names should go in the /data folder of darknet. Don't forget to create the train.txt and test.txt that inria.data calls.

Finally, start training using the following command:

$ ./darknet -i 0 detector train data/inria.data cfg/yolov3_inria.cfg darknet53.conv.74

Nevertheless, you can use the RunNonStop script, if you sometimes you training stops because of insufficient memory.

Demos

Check some results with Caltech Pedestrian dataset (videos)

Test1 Test2

About

Pedestrian Detection using Caltech and INRIA dataset with Darknet Framework.


Languages

Language:Jupyter Notebook 96.8%Language:Python 2.5%Language:Shell 0.8%