项目作者: Aramrt

项目描述 :
Training YoloV3 with custom dataset
高级语言: C++
项目地址: git://github.com/Aramrt/YOLOV3-custom-dataset.git
创建时间: 2021-09-04T08:00:51Z
项目社区:https://github.com/Aramrt/YOLOV3-custom-dataset

开源协议:

下载


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)