----制作自己的数据集----

1、下载labelimg（可在网页自行下载）

---

2、利用labelimg进行图片标注    

（1）修改data/predefined_classes标签 例如我需要标注两类“smoke” “fire”    👇

 

---

（2）新建一个"Mydate"文件夹存放自己的数据  将需要标注的图片导入Mydate/images/train 👇

          

---

（3）运行labelimg.py

    （首先会出现一个黑框 不用理会 等待程序打开之后可直接使用）

 

          ① 点击上侧任务栏的“View”并选中“Aotu Save mode”（自动保存 ）👇

         

          ② 将标签文件模式改为YOLO 👇           （YOLO格式的数据在标签部分生成的是txt文件）

 

                                                   👉                

 

          ③  文件夹选择

               -1- Open Dir 选择需要标注的图片所在文件夹 Mydate/images/train   👇         

               -2- Change Save Dir 选择标注后的标签存放的文件夹 Mydate/labels/train  👇

                                                                    

               

                    选择后文件夹内所有图片会显示在右下角  👇

                                   

 

          ④ 开始标注 

               -1- 点击 Create\nRectBox 或者快捷键“w”框出所标注的内容并选择对应的标签   👇             

 

               -2- 再点 Next Image 或“d”进行下一张图片的标注直至全部标注完成 👇

                                                                              

 

               ------标注完成后会生成与图片名称对应的txt文件  👇
 

---

 

 一些快捷键：

                                “w”标注画框

                                “a”上一张图片

                                “d”下一张图片

                                “del”删除画的框

                                “←↑↓→”移动框的位置

---

以上就是制作数据集的方法了，制作完成后就可以进行训练啦