-
Notifications
You must be signed in to change notification settings - Fork 120
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #845 from Panchadip-128/Added-Object-Detection-usi…
…ng-COCO-Dataset-with-the-help-of-advanced-YOLO_v5-model Added object detection using coco dataset with the help of advanced yolo v5 model
- Loading branch information
Showing
6 changed files
with
684 additions
and
0 deletions.
There are no files selected for viewing
603 changes: 603 additions & 0 deletions
603
...O dataset with YOLO_v5 advanced model/Object Detection with COCO_dataset_using YOLO.ipynb
Large diffs are not rendered by default.
Oops, something went wrong.
26 changes: 26 additions & 0 deletions
26
...ision/Object detection using COCO dataset with YOLO_v5 advanced model/README.md
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,26 @@ | ||
# Project Title: Object Detection with YOLOv5 | ||
|
||
Project Description: | ||
-------------------- | ||
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. | ||
|
||
Features: | ||
--------- | ||
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 |
1 change: 1 addition & 0 deletions
1
...ision/Object detection using COCO dataset with YOLO_v5 advanced model/coco_img_YOLO_model
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1 @@ | ||
|
38 changes: 38 additions & 0 deletions
38
...uter Vision/Object detection using COCO dataset with YOLO_v5 advanced model/gitignore.txt
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,38 @@ | ||
__pycache__/ | ||
*.py[cod] | ||
*.so | ||
|
||
.ipynb_checkpoints/ | ||
|
||
# Python virtual environments | ||
env/ | ||
venv/ | ||
|
||
# YOLOv5 Weights | ||
*.pt | ||
runs/ | ||
weights/ | ||
|
||
|
||
*.log | ||
|
||
|
||
*.o | ||
*.a | ||
*.out | ||
*.exe | ||
|
||
# Data files | ||
data/ | ||
*.csv | ||
*.json | ||
|
||
# System files | ||
.DS_Store | ||
Thumbs.db | ||
|
||
# PyCharm/IDEA | ||
.idea/ | ||
|
||
# Visual Studio Code | ||
.vscode/ |
Binary file added
BIN
+106 KB
...ect detection using COCO dataset with YOLO_v5 advanced model/obj-det-output.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
16 changes: 16 additions & 0 deletions
16
...r Vision/Object detection using COCO dataset with YOLO_v5 advanced model/requirements.txt
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,16 @@ | ||
torch>=1.7.0 | ||
torchvision>=0.8.0 | ||
matplotlib>=3.2.2 | ||
numpy>=1.18.5 | ||
opencv-python>=4.1.2 | ||
pillow>=7.1.2 | ||
PyYAML>=5.3.1 | ||
tqdm>=4.64.0 | ||
scipy>=1.4.1 | ||
tensorboard>=2.4.1 | ||
seaborn>=0.11.0 | ||
pandas>=1.1.4 | ||
albumentations>=0.5.2 | ||
ipython>=7.16.1 | ||
jupyterlab>=2.1.5 | ||
requests>=2.23.0 |