Skip to content

In this repository, I aim at providing theoretical and practical notes for fully understanding Yolo models. Then, I show how to label a dataset which is downloaded from kaggle.com using makesense.ai to make it ready for training by yolo models

License

Notifications You must be signed in to change notification settings

pooya-mohammadi/intro_to_yolo

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Intro to Yolo

yolo-img
In this repository, I aim at providing theoretical and practical notes for fully understanding Yolo models. Then, I show how to label a dataset which is downloaded from kaggle.com using makesense.ai to make it ready for training by yolo models. In addition, I have prepared a series of YouTube videos to make the process of learning and training a custom dataset as smooth and easy as possible. You can find the YouTube videos in the following playlist:
Intro_to_Yolo YouTube videos

Table of contents

Downloading dataset

The dataset is available on kaggle.com in the following link:
Car License Plate Detection
By clicking on the download link the download begins. kaggle_download

Note: It may ask you to log in with your kaggle account, the simplest way is to log in with one the third-parties likes Google Account.

Labeling dataset

For labeling this dataset, I use makesense.ai. It's a handy and easy-to-begin system for labeling different image-based datasets. To better understand how to use makesense, I recorded a video which you can find in the following link:
Makesense.ai YouTube video

First step: Open makesense.ai main page
second step: clickGet Started Then Select images by dropping or from your folder and choice .
get started
drop_or_choice_file
clik_object_detection

Third step: Before starting, you must choose a name for your label (you can have two or more labels and add a new chosen name by pressing the enter button).
name_lable

Fourth step: Based on the type of labeling, choose the mode you want (for example, we have chosen Rect) and then drag the label you want on the image.
the_mode
dragging_the_tag

Fifth step: Be sure to select a name of your choice after dragging the tag (if there are two tags on the page, both tags must be named).
name_label_choice

Sixth: After completing the labels, we select the Export Annotation option for the output from the action.(You can get any output model you need)
annotation
model_output

🤓How to Extract

right-click on the downloaded file and select the extract file option, the labels will be extracted as a file Last step: Checking the labels extract
extract_as_file

💡 Re-labeling or completing incomplete labels

First step: Create a text file named labels.txt in your labels' folder. Put the names you chose for your label in order (be careful in the order of the names)
create_label_txt
name_in_label_tag
Second step: Reload the photos and instead of selecting the name of the label, select the start project option.
choice_start_project
Third step: Select the import annotation option in Action and drag or select the labels along with the created "labels.txt" (Make sure that the type of import is the same as the type of export).
choice_model_import
import_annotation
import_succefilly
Congratulations, you can now continue or improve your project :) .
complete_again

Training a model with Yolov5

from deep_utils import YOLOV5TorchObjectDetector
YOLOV5TorchObjectDetector.test_label_dir()

🌟 Spread the word!

I would appreciate it if you could support the active development of this repo by:

Thank you so much for your interest in growing the reach of the repo!

(back to top)

⚠️ License

Distributed under the MIT License. See LICENSE for more information.

(back to top)

References:

🍀 deep_utils: https://github.com/pooya-mohammadi/deep_utils

🍀 yolov5: https://github.com/pooya-mohammadi/yolov5-gradcam

🍀 makesense: https://www.makesense.ai/

About

In this repository, I aim at providing theoretical and practical notes for fully understanding Yolo models. Then, I show how to label a dataset which is downloaded from kaggle.com using makesense.ai to make it ready for training by yolo models

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published