Object Specifier (Yolo-Coco)

codewdhruv / obj-specifier

AI model to detect objects from a given input picture

Object Specifier

Dataset

The dataset used can be downloaded here - Click to Download

This is ready to use data with weights and configuration along with coco names to detect objects with YOLO algorithm.

Prerequisites

All the dependencies and required libraries are included in the file requirements.txt See here

Installation

1. Clone the repo

$ git clone https://github.com/dhrubajyoti89/obj-specifier.git

2. Change your directory to the cloned repo and create a Python virtual environment named 'test'

$ mkvirtualenv test

3. Now, run the following command in your Terminal/Command Prompt to install the libraries required

$ pip3 install -r requirements.txt

Working

1. Open terminal. Go into the cloned project directory folder and type the following command:

$ python3 yolo.py --dataset dataset

2. To detect objects in an image type the following command:

$ python3 yolo.py -i "images/1.jpg" -y "yolo-coco"

Results

Our model gave 81% accuracy for Object Detection after training

Project Demo

Dev Link

Feel free to mail…

View on GitHub

Built it using yolo based object detector model. The dataset was made custom using object recognition API and Kaggle datasets of real large datasets of object images(yolo-coco). Used Python script with numpy, argparse, os libraries as the backend.