O'Reilly logo

Learn OpenCV 4 by Building Projects - Second Edition by Prateek Joshi, Vinicius G. Mendonca, David Millan Escriva

Stay ahead with the world's most comprehensive technology and business learning platform.

With Safari, you learn the way you learn best. Get unlimited access to videos, live online training, learning paths, books, tutorials, and more.

Start Free Trial

No credit card required

The YOLO dataset, vocabulary, and model

Before we start to import the model into our OpenCV code, we have to obtain it through the YOLO website: https://pjreddie.com/darknet/yolo/. This provides pre-trained model files based on the COCO dataset, which contains 80 object categories, such as person, umbrella, bike, motorcycle, car, apple, banana, computer, and chair.

To get all the names of categories and uses for visualization, check out https://github.com/pjreddie/darknet/blob/master/data/coco.names?raw=true.

The names are in the same order as the results of deep learning model confidences. If you want to see some images of the COCO dataset by category, you can explore the dataset at http://cocodataset.org/#explore, and download some of them ...

With Safari, you learn the way you learn best. Get unlimited access to videos, live online training, learning paths, books, interactive tutorials, and more.

Start Free Trial

No credit card required