Custom dataset

[vaibhavkumar049]

How to train model on custom dataset?

[mixy1]

hey I'd suggest trying to use the alexeyab darknet repository to use yolo. we ended up switching two times first this repository then opencv now darknet. also using darknet gives vastly superior performance on a cuda system. worse performance on a cpu system though but easier training.

[Ramanujam-RS]

@mixy1 Why is that you switched from this YOLO to alexyab?

[mixy1]

well AlexeyAB's yolo is still mantained. also we needed to change the neural network a bit and this yolo doesn't allow for that.