danieladejumo/darknet-coco-object_detection

Darknet Object Detection on the COCO dataset

This model uses a pretrained YOLO Darknet model to perform object detection on an input image. The model is able to identify 80 classes from the COCO dataset. The classes are listed here config/coco.names.

Usage

Clone the repository using

repo = Repository("/local_repo_name", clone_from="danieladejumo/darknet-coco-object_detection")

Run a detection by using the function detect(path_to_image) in the notebook darknet-coco-object_detection.ipynb. The output image with the detection rectangle and classes will be saved to images/image_file_name-det.jpg