yolov5_violence_detect

KO | EN

gitlite — search

#python #java #python3 #arduino #golang #machine-learning #rust #html #flask #javascript #seismology #nodejs

yolov5_violence_detect

★ 18 Open GitHub ↗

基于yolov5s进行的暴力行为检测

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

## 一. 模型修改

### 1. 使用模型

* 本次训练使用了 yolov5s 模型，所以需要对部分配置文件进行修改：

> 需要对位于文件夹内 ./models/yolov5s.yaml 进行修改，训练修改了其的复制文件：
  >
  > ![image-20240702130507391](https://s2.loli.net/2024/07/02/KqcBmOxFQCH8Jb5.png)
  >
  > 根据自身数据集标签数量对 nc 进行修改。
  >
  > 同时，对 ./data/train.yaml 进行修改：
  >
  > ![image-20240702130935631](https://s2.loli.net/2024/07/02/beSxkNKn15zEvAC.png)   
  >
  > 最后修改训练文件：
  >
  > ![image-20240702131035186](https://s2.loli.net/2024/07/02/CIa5EpGHYTudn8w.png)
  >
  > 并且调整其中参数。

## 二. 训练数据

## 1. 性能指标分析

### (1) 模型数据

> 训练好的模型保存在 ./runs/train/exp2/weights 中
>
> 其分析数据存放于同级文件夹下
>
> ![image-20240702131927555](https://s2.loli.net/2024/07/02/cyBMFrLbG7K3tSx.png)

### (2). 数据分析

* lables 反映了训练集的数据量，以及其图片和标签的部分参数

> ![image-20240702132751484](https://s2.loli.net/2024/07/02/tXIApZGYnuvkQcM.png)
  >
  > ![image-20240702132812189](https://s2.loli.net/2024/07/02/5U61xiECcqtLove.png)

* results 反映了训练集和数据集的结果，包含了：训练次数，GPU消耗，边界框损失，目标检测损失，total，targets, 图片大小，P，R，mAP@.5, mAP@.5:.95, 验证集val Box, 验证集val obj, 验证集val cls

> ![image-20240702132439965](https://s2.loli.net/2024/07/02/1iyWOg9sJeaVmFx.png)
>
> 上图为 train.csv

同时，也可以通过损失函数评估模型性能。

- 定位损失box_loss：预测框与标定框之间的误差（GIoU）

- 置信度损失obj_loss：计算网络的置信度

- 分类损失cls_loss：计算锚框与对应的标定分类是否正确

> ![image-20240702132925434](https://s2.loli.net/2024/07/02/szFiV4IHckthapw.png)
  >
  > 上图为train.png

**特别注意到：在最后数轮次的训练中，验证集的损失有所上升，可能需要更换置信度损失**

其他曲线趋于稳定，**cls_loss则始终为0，可能是class设置不多，模型训练未将其丢失**

### (3). 模型构成

于pycharm终端 ，输入 `tensorboard --logdir runs`，将结果可视化

> ![image-20240702134122722](https://s2.loli.net/2024/07/02/YOgtkU4XfnQszJR.png)

通过浏览器查看可视化训练结果

> ![image-20240702134256003](https://s2.loli.net/2024/07/02/FBYMPhvZowIDJRr.png)

于其中查看本次模型的构成：

> ![image-20240702134339607](https://s2.loli.net/2024/07/02/eFSCuI5novXAQgk.png)
>
> ![train_exp2](https://s2.loli.net/2024/07/02/P7HqmkEtNQW51Xb.png)
>
> 可以看到，模型通过多层卷积，采样，实现训练。
>
> ![image-20240702134652066](https://s2.loli.net/2024/07/02/cUXPRBNlT2YOAVw.png)
>
> 上图yolov5s的代码部分

## 训练结果

模型将会检测视频或图片中的暴力行为

![image-20240702140535731](https://s2.loli.net/2024/07/02/wlYWRk2PbcKLtMG.png)

![image-20240702140509069](https://s2.loli.net/2024/07/02/djh47VrcXKn1SNe.png)

![image-20240702140419713](https://s2.loli.net/2024/07/02/4aAWEDNBbd3CHrV.png)

![image-20240702140610667](https://s2.loli.net/2024/07/02/q68iygr2kTYK3pJ.png)

**注意到： 模型在面对人体剧烈姿态时，可能会出现假正例，以及部分假负例。可能是模型对人体姿态和二者距离之间产生错误联系。需要对数据集标签重新精确标定，并且加大数据集训练**

yolov5_violence_detect

Explore Similar Repositories

Was this content helpful?

Recent Feedback

Download README