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