基于YOLOv8的暴力行为分类系统

本项目旨在通过计算机视觉技术实现暴力行为的实时监测和分类。利用先进的目标检测算法YOLOv8,结合Streamlit开发的前端界面,该系统能够高效地从图像或视频中识别出暴力行为,并及时报警。系统支持对实时摄像头输入进行分析,以及上传的图像和视频文件。

技术栈:
  • YOLOv8: 目标检测模型,用于检测图像中的物体。
  • Streamlit: Python Web应用框架,用于构建用户界面。
功能特点:
  • 支持图片、视频及实时摄像头输入。
  • 实现暴力行为与正常行为的二分类。
  • 提供直观的用户界面,便于操作和监控。
混淆矩阵示例(假设数据):
复制代码
          预测值
          正常  暴力
实际值
正常   95    5
暴力   10   85

这表明模型在测试集上具有较高的准确率,其中95%的正常行为被正确分类为正常,而85%的暴力行为被正确识别。

关键代码示例

下面是一些关键代码片段,包括如何使用YOLOv8进行预测以及如何用Streamlit构建用户界面。

YOLOv8预测代码

首先确保安装了ultralytics库:

复制代码
pip install ultralytics

然后编写预测函数:

复制代码
from ultralytics import YOLO

def predict(image_path, model):
    """
    使用YOLOv8模型进行预测。
    :param image_path: 图像路径
    :param model: 加载的YOLOv8模型
    :return: 预测结果
    """
    results = model.predict(image_path)
    return results[0].boxes.cls
Streamlit前端代码

创建一个简单的Web应用:

复制代码
import streamlit as st
from PIL import Image
import torch

# 加载模型
model = YOLO('path/to/yolov8_model.pt')

# Streamlit应用
st.title("暴力行为分类系统")

# 文件上传
uploaded_file = st.file_uploader("上传图片或视频", type=["jpg", "jpeg", "png", "mp4"])

if uploaded_file is not None:
    # 保存上传的文件到临时目录
    temp_file_path = f"temp_{uploaded_file.name}"
    with open(temp_file_path, mode='wb') as f:
        f.write(uploaded_file.read())

    # 显示上传的文件
    if uploaded_file.name.endswith(('jpg', 'jpeg', 'png')):
        image = Image.open(uploaded_file)
        st.image(image, caption='Uploaded Image.', use_column_width=True)
    elif uploaded_file.name.endswith('mp4'):
        st.video(uploaded_file)

    # 进行预测
    predictions = predict(temp_file_path, model)

    # 输出预测结果
    if 0 in predictions:
        st.write("预测结果: 正常")
    elif 1 in predictions:
        st.write("预测结果: 暴力")
    else:
        st.write("无法识别的行为")

注意事项

  • 上述代码需要根据实际情况调整模型路径等细节。
  • 对于实时摄像头的支持,您可以使用OpenCV来捕获视频流并将其传递给预测函数。
  • 在部署前,请确保所有依赖项都已经安装并且兼容。
相关推荐
铭keny5 分钟前
YOLO11 目标检测从安装到实战
人工智能·目标检测·目标跟踪
杨小扩6 小时前
第4章:实战项目一 打造你的第一个AI知识库问答机器人 (RAG)
人工智能·机器人
whaosoft-1436 小时前
51c~目标检测~合集4
人工智能
雪兽软件6 小时前
2025 年网络安全与人工智能发展趋势
人工智能·安全·web安全
元宇宙时间6 小时前
全球发展币GDEV:从中国出发,走向全球的数字发展合作蓝图
大数据·人工智能·去中心化·区块链
小黄人20257 小时前
自动驾驶安全技术的演进与NVIDIA的创新实践
人工智能·安全·自动驾驶
ZStack开发者社区8 小时前
首批 | 云轴科技ZStack加入施耐德电气技术本地化创新生态
人工智能·科技·云计算
X Y O9 小时前
神经网络初步学习3——数据与损失
人工智能·神经网络·学习
FL16238631299 小时前
如何使用目标检测深度学习框架yolov8训练钢管管道表面缺陷VOC+YOLO格式1159张3类别的检测数据集步骤和流程
深度学习·yolo·目标检测
唯创知音9 小时前
玩具语音方案选型决策OTP vs Flash 的成本功耗与灵活性
人工智能·语音识别