AlexeyAB / darknet

YOLOv4 / Scaled-YOLOv4 / YOLO - Neural Networks for Object Detection (Windows and Linux version of Darknet )
http://pjreddie.com/darknet/
Other
21.71k stars 7.96k forks source link

Deploy the trained model in production #2677

Open DeeeepNet opened 5 years ago

DeeeepNet commented 5 years ago

Hi @AlexeyAB, Please guide me if I want to train yolo3 for custom my own dataset on ubuntu 16.04 and deploy the trained model on server or GPU or embedded system that have Linux based OS. I want to load the trained model on GPU or RAM of the system only once and then every time the camera captured the image feed to the loaded model. Is enough the one 1080-ti GPU for training yolo3?

thanks in advance.

AlexeyAB commented 5 years ago

@DeeeepNet Hi,

Please guide me if I want to train yolo3 for custom my own dataset on ubuntu 16.04 and deploy the trained model on server or GPU or embedded system that have Linux based OS.

So, just compile Darknet with GPU=1 CUDNN=1 OPENCV=1 LIBSO=1 and use https://github.com/AlexeyAB/darknet#how-to-compile-on-linux

Is enough the one 1080-ti GPU for training yolo3?

Usually yes. Just train while mAP is increasing: https://github.com/AlexeyAB/darknet#when-should-i-stop-training