Yolov5 label and establish your own data set

Welcome to the world of machine learning  
Blog home page ： Qingyun Pavilion

Welcome to focus on the likes collection ️ Leaving a message.

This article was originally written by Qing Yunge ！

This stage belongs to the Qi training stage , I hope you can make a breakthrough

Starting time ：2021 year 7 month 16 Japan

️ I hope I can complete the advanced road with you ！

The author's level is very limited , If an error is found , Please leave a message ！ Thank you very much ！

---

labelImg download

    labelImag stay github It has , You can download it yourself
Download link : labelImg-master.

1. Modify the label
    stay data/predefined_classes Modify the inside class, For example, my label is red and green Just change it in this format .

 

2. Then click directly labelImg.py file open
    The operating environment generally includes python Of idle Can be opened directly , If it is pycharm Will be in pycharm Open the code , Then manually run, At this time, kids may encounter this problem . 

conda install pyqt=5
conda install -c anaconda lxml
pyrcc5 -o libs/resources.py resources.qrc
python labelImg.py
python labelImg.py [IMAGE_PATH] [PRE-DEFINED CLASS FILE]

Home page information tells us , When we download the source code , First of all Anaconda Prompt Enter the unzipped file directory , Execute the above instructions in turn . The first three instructions are indispensable （ Unless the first two... Are already installed in the environment ）, The third instruction is the most important ! 

 

3. After opening, change the format to yolo

4. Choose the location of the picture and what we marked txt Where to save the file
      It's usually used labelImg We all need to create in the file images The folder and labels Folder , Put the pictures to be marked on images In the folder ,labels Don't put things in the folder for the time being , Annotations will be automatically generated at that time txt File and Open Dir choice images,Change Save Dir choice labels.
The following figure shows where my dataset is saved  

5. Then we can start to label our dataset
Press “w” Key can start annotation
Press “a” The key is the previous one
Press “d” Key is next

6. Save automatically
      If we need to quickly label data sets , Can be in view/atuo save mode Select auto save , In this way, the selection box of whether to save or not will not pop up when you press one . 

7. Faster operation
      If many sheets are marked red, You can put "use default label" Check on , Fill in the back red.