UppuluriKalyani / ML-Nexus

ML Nexus is an open-source collection of machine learning projects, covering topics like neural networks, computer vision, and NLP. Whether you're a beginner or expert, contribute, collaborate, and grow together in the world of AI. Join us to shape the future of machine learning!
https://ml-nexus.vercel.app/
MIT License
69 stars 122 forks source link

Feature request: Object Detection using COCO (refined) dataset with Advanced YOLO-v5 model #840

Closed Panchadip-128 closed 2 weeks ago

Panchadip-128 commented 2 weeks ago

Is your feature request related to a problem? Please describe.

This project implements an object detection system using the YOLOv5 (You Only Look Once) model. YOLOv5 is a state-of-the-art, real-time object detection algorithm that is both fast and accurate. This system can detect multiple objects in images or video streams and can be further fine-tuned for custom datasets. It includes training the YOLOv5 model, evaluating it on a test dataset, and running real-time inference.

Describe the solution you'd like Real-time object detection on images and video streams. Training the YOLOv5 model on custom datasets. Evaluation using key metrics such as Precision, Recall, Intersection over Union (IoU), and Mean Average Precision (mAP). Deployment for detecting objects in images and video streams (GPU requirements for this case is much preferrable) Model robustness testing with image augmentations.

Requirements: Python 3.7+ PyTorch 1.7+ YOLOv5 (via the ultralytics/yolov5 repository) Common libraries: numpy opencv-python torch pillow matplotlib albumentations

Describe alternatives you've considered A clear and concise description of any alternative solutions or features you've considered.

Approach to be followed (optional) A clear and concise description of the approach to be followed.

Additional context Add any other context or screenshots about the feature request here.

github-actions[bot] commented 2 weeks ago

Thanks for creating the issue in ML-Nexus!🎉 Before you start working on your PR, Pull the latest changes to avoid any merge conflicts.

github-actions[bot] commented 2 weeks ago

Hello @Panchadip-128! Your issue #840 has been closed. Thank you for your contribution!