Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/sanu0711/object-detection-using-the-yolo-model

The objective of this project is to demonstrate the implementation of object detection using the YOLO model, transformers library, and OpenCV. The provided Python script utilizes a pre-trained YOLO model (hustvl/yolos-tiny) for detecting objects in images.
https://github.com/sanu0711/object-detection-using-the-yolo-model

bounding-boxes huggingface-transformers jupyter-notebook matplotlib object-detection opencv-python pil python3 transformer yolo yolov4-tiny yolov5

Last synced: 3 days ago
JSON representation

The objective of this project is to demonstrate the implementation of object detection using the YOLO model, transformers library, and OpenCV. The provided Python script utilizes a pre-trained YOLO model (hustvl/yolos-tiny) for detecting objects in images.

Awesome Lists containing this project

README

        

# Object Detection using YOLO Model

![GitHub last commit](https://img.shields.io/github/last-commit/sanu0711/Object-Detection-using-the-YOLO-model)
![GitHub repo size](https://img.shields.io/github/repo-size/sanu0711/Object-Detection-using-the-YOLO-model)

## Overview
This project showcases object detection using the YOLO (You Only Look Once) model in Python, utilizing the transformers library for integration and OpenCV for image processing. The YOLO model is a state-of-the-art algorithm for real-time object detection tasks.

## Objective
The objective of this project is to demonstrate the implementation of object detection using the YOLO model, transformers library, and OpenCV. The provided Python script utilizes a pre-trained YOLO model (`hustvl/yolos-tiny`) for detecting objects in images.

## Features
- Developed a Python script for object detection using the YOLO model.
- Utilized the transformers library for YOLO model integration.
- Incorporated image processing techniques with the PIL and OpenCV libraries.
- Integrated pre-trained YOLO model (`hustvl/yolos-tiny`) for object detection tasks.
- Processed input images and generated bounding boxes using the YolosImageProcessor.
- Demonstrated the detection results with bounding boxes overlaid on the original image using matplotlib.
- Successfully detected objects with high confidence levels (>90%) and displayed the results visually.

## Installation
1. Clone this repository to your local machine.
2. Install the required dependencies/library.

## Results
![Object Detection Screenshot1](https://github.com/sanu0711/Object-Detection-using-the-YOLO-model/blob/main/Results/Screenshot%202024-05-29%20134831.png)

![Object Detection Screenshot2](https://github.com/sanu0711/Object-Detection-using-the-YOLO-model/blob/main/Results/Screenshot%202024-05-29%20134859.png)
![Object Detection Screenshot3](https://github.com/sanu0711/Object-Detection-using-the-YOLO-model/blob/main/Results/Screenshot%202024-05-29%20134918.png)

## References
- [Ultralytics GitHub Repository](https://github.com/ultralytics/yolov5)

## Contributors

- [Abhishek Yadav](https://github.com/sanu0711)