Skip to content

Aramrt/YOLOV3-custom-dataset

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

14 Commits
 
 
 
 
 
 

Repository files navigation

YOLOV3-custom-dataset

dataset:

Surf makesense for annotating your pictures.

  1. try to grab your boxes a little bigger than the objects.

  2. export annotation as textes.

Train YOLOV3 on your custom dataset (follow the structure):

if you want to train yolov3 on google colab you don't need to download cuda, cudnn and opencv.

training yolov3 on google colab --> YOLOV3-COLAB

Local PC:

Download CUDA and CUDNN based on your computer hardware and OpenCV Versions.

surf this link for building OpenCV GPU

surf this link for building Darknet.

Try to run your weight in CPP

My GPU : 1050Ti

10 FPS in python

25 FPS in cpp

Warning: change the weight and cfg configuration to CUDA (change the CPU and OPENCV to CUDA)