Importing YOLO

In this section, we will create an instance of YOLO classes. Note that it may take a little time as the YOLO model needs to load up:

yolo = YOLO(0.6, 0.5)
file = 'data/coco_classes.txt'
all_classes = get_classes(file)

In the next section, we will test the implementation of the YOLO model by predicting objects in an image, as well as a video.