基于Yolov5的吸烟行为识别检测_吸烟数据集

---

前言

  上一篇《win10系统下Yolov5目标检测环境搭建（Anaconda3+Pytorch+Yolov5，CPU，无GPU）》中，介绍了win10系统下yolov5的环境搭建步骤，本篇在此环境下，实现吸烟行为识别检测。

一、数据集smoke_data

1、Annotations文件夹

存放labelimg标注工具生成的.xml标注文件，每一张图片对应一个xml文件

2、ImageSets/Main文件夹

存放train.txt，val.txt，test.txt和trainval.txt文件

3、JPEGImages文件夹

存放待训练的图片文件，我使用的所有图片都为jpg格式

二、数据集划分

新建split_train_val.py文件，其作用主要是生成train.txt，val.txt，test.txt和trainval.txt文件，生成的这些文件，存放在ImageSets/Main文件夹下。py脚本内容如下：

# coding:utf-8
 
import os
import random
import argparse
 
parser = argparse.ArgumentParser()
#xml文件的地址，根据自己的数据进行修改 xml一般存放在Annotations下
parser.add_argument('--xml_path', default='smoke_data/Annotations', type=str, help='input xml label path')
#数据集的划分，地址选择自己数据下的ImageSets/Main
parser.add_argument('--txt_path', default='smoke_data/ImageSets/Main', type=str, help='output txt label path')
opt = parser.parse_args()
 
trainval_percent = 0.9
train_percent = 0.9
xmlfilepath = opt.xml_path
txtsavepath = opt.txt_path
total_xml = os.listdir(xmlfilepath)
if not os.path.exists(txtsavepath):
    os.makedirs(txtsavepath)
 
num = len(total_xml)
list_index = range(num)
tv = int(num * trainval_percent)
tr = int(tv * train_percent)
trainval = random.sample(list_index, tv)
train = random.sample(trainval, tr)
 
file_trainval = open(txtsavepath + '/trainval.txt', 'w')
file_test = open(txtsavepath + '/test.txt', 'w')
file_train = open(txtsavepath + '/train.txt', 'w')
file_val = open(txtsavepath + '/val.txt', 'w')
 
for i in list_index:
    name = total_xml[i][:-4] + '\n'
    if i in trainval:
        file_trainval.write(name)
        if i in train:
            file_train.write(name)
        else:
            file_val.write(name)
    else:
        file_test.write(name)
 
file_trainval.close()
file_train.close()
file_val.close()
file_test.close()
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

执行命令，如下：

python split_train_val.py
1

可以看到上述txt文件，已经生成，如下：

三、使用labelimg工具标注

进入之前建立的pytorch环境，安装labelimg，命令如下：

pip install labelimg
1

输入labelimg命令，启动，如下：

选择使用“PascalVOC”格式；
点击“open dir”按钮，选择JPEGImages文件夹目录；
点击“change save dir”按钮，选择Annotations文件夹目录；
点击“create RectBox”按钮，选择区域进行标注；
标注完成后，点击“save”按钮，进行保存，xml会存放在Annotations文件夹下。

四、将数据集格式转化成yolo_txt格式

  即将xml文件转化成txt文件，每一张图片对应一个txt文件，文件中每一行对应一个目标信息，包括class, x_center, y_center, width和height。
  新建voc_label.py脚本，将classes 里面的内容改成自己的类别，若是多个类别，则用逗号分隔，内容如下：

# -*- coding: utf-8 -*-
import xml.etree.ElementTree as ET
import os
from os import getcwd
 
sets = ['train', 'val', 'test']
classes = [
    "smoke"]   # 改成自己的类别
abs_path = os.getcwd()
print(abs_path)
 
def convert(size, box):
    dw = 1. / (size[0])
    dh = 1. / (size[1])
    x = (box[0] + box[1]) / 2.0 - 1
    y = (box[2] + box[3]) / 2.0 - 1
    w = box[1] - box[0]
    h = box[3] - box[2]
    x = x * dw
    w = w * dw
    y = y * dh
    h = h * dh
    return x, y, w, h
 
def convert_annotation(image_id):
    in_file = open('F:/SVN-ZJKY/YiFeiShouJiRobot/yolov5-master/smoke_data/Annotations/%s.xml' % (image_id), encoding='UTF-8')
    out_file = open('F:/SVN-ZJKY/YiFeiShouJiRobot/yolov5-master/smoke_data/labels/%s.txt' % (image_id), 'w')
    tree = ET.parse(in_file)
    root = tree.getroot()
    size = root.find('size')
    w = int(size.find('width').text)
    h = int(size.find('height').text)
    for obj in root.iter('object'):
        difficult = obj.find('difficult').text
        # difficult = obj.find('Difficult').text
        cls = obj.find('name').text
        if cls not in classes or int(difficult) == 1:
            continue
        cls_id = classes.index(cls)
        xmlbox = obj.find('bndbox')
        b = (float(xmlbox.find('xmin').text), float(xmlbox.find('xmax').text), float(xmlbox.find('ymin').text),
             float(xmlbox.find('ymax').text))
        b1, b2, b3, b4 = b
        # 标注越界修正
        if b2 > w:
            b2 = w
        if b4 > h:
            b4 = h
        b = (b1, b2, b3, b4)
        bb = convert((w, h), b)
        out_file.write(str(cls_id) + " " + " ".join([str(a) for a in bb]) + '\n')
 
wd = getcwd()
for image_set in sets:
    if not os.path.exists('F:/SVN-ZJKY/YiFeiShouJiRobot/yolov5-master/smoke_data/labels/'):
        os.makedirs('F:/SVN-ZJKY/YiFeiShouJiRobot/yolov5-master/smoke_data/labels/')
    image_ids = open('F:/SVN-ZJKY/YiFeiShouJiRobot/yolov5-master/smoke_data/ImageSets/Main/%s.txt' % (image_set)).read().strip().split()
    list_file = open('smoke_data/%s.txt' % (image_set), 'w')
    for image_id in image_ids:
        list_file.write(abs_path + '/smoke_data/images/%s.jpg\n' % (image_id))
        convert_annotation(image_id)
    list_file.close()
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62

执行命令，如下：

python voc_label.py
1

运行该脚本，会生成labels文件夹和三个包含数据集的txt文件，即该脚本的作用：将训练集、验证集、测试集生成label标签（训练中要用到），同时将数据集路径导入txt文件中。如下：

其中，labels文件夹，存放xml文件转化后的txt文件；
train.txt文件，记录的是训练集图像绝对路径；
test.txt文件，记录的是测试集图像绝对路径；
val.txt文件，记录的是验证集图像绝对路径；

五、增加自定义数据集文件

在yolov5-master/data/文件夹下，新建smoke_data.yaml文件，内容如下：

train: F:/SVN-ZJKY/YiFeiShouJiRobot/yolov5-master/smoke_data/train.txt
val: F:/SVN-ZJKY/YiFeiShouJiRobot/yolov5-master/smoke_data/val.txt
test: F:/SVN-ZJKY/YiFeiShouJiRobot/yolov5-master/smoke_data/test.txt
 
# Classes
nc: 1  # number of classes
names: ['smoke']  # class names
1
2
3
4
5
6
7

注意：修改自己对应的类别个数和类别名。

六、修改模型文件yolov5s.yaml

我选择使用的是yolov5s模型，进入yolov5-master/model/文件夹下，找到yolov5s.yaml文件，修改对应的类别个数nc，如下：

# YOLOv5 
声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/繁依Fanyi0/article/detail/588244