[python]基于yolov12实现热力图可视化支持图像视频和摄像头检测

YOLOv12 Grad-CAM 可视化工具

本工具基于YOLOv12模型,结合Grad-CAM技术实现目标检测的可视化分析,支持图像、视频和实时摄像头处理。

注意

该项目使用的是yolov12-1.0模型进行测试通过,不是使用turbo模型,且由于yolov12-1.0由于更新目前直接不支持显卡RTX2080及其之前的显卡了,运行会提示报错。请使用我这个yolov12早期更新版本,此版本是支持RTX2070显卡的

功能特性

  • 支持多种Grad-CAM方法
  • 实时摄像头处理
  • 视频文件处理
  • 图像文件处理

环境要求

  • Python 3.10+
  • 需要电脑带有nvidia显卡,且有cuda环境支持

安装步骤

  1. 创建虚拟环境安装yolov12(推荐)
bash 复制代码
wget https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.3/flash_attn-2.7.3+cu11torch2.2cxx11abiFALSE-cp311-cp311-linux_x86_64.whl
conda create -n yolov12 python=3.11
conda activate yolov12
pip install -r requirements.txt
pip install -e .

注意上述是linux安装步骤,windows可以参考https://blog.csdn.net/FL1623863129/article/details/145760810安装

  1. 克隆仓库
bash 复制代码
下载源码yolov12-gradcam
cd yolov12-gradcam
  1. 安装依赖
bash 复制代码
pip install -r requirements.txt

快速开始

1. 图像处理

python 复制代码
from Yolov12Heatmap import *

model = Yolov12Heatmap(weight='yolov12n.pt')
result = model('test.jpg')
cv2.imshow('result', result)
cv2.waitKey(0)

2. 视频处理

python 复制代码
from Yolov12Heatmap import *

model = Yolov12Heatmap(weight='yolov12n.pt')
model.start_video('test.mp4')

3. 实时摄像头

python 复制代码
from Yolov12Heatmap import *

model = Yolov12Heatmap(weight='yolov12n.pt')
model.start_camera(0)  # 0表示默认摄像头

参数配置

更多参数调节需要读懂Yolov12Heatmap.py文件代码进行修改

相关推荐
烛阴4 小时前
简单入门Python装饰器
前端·python
要努力啊啊啊4 小时前
YOLOv3-SPP Auto-Anchor 聚类调试指南!
人工智能·深度学习·yolo·目标检测·目标跟踪·数据挖掘
好开心啊没烦恼4 小时前
Python 数据分析:numpy,说人话,说说数组维度。听故事学知识点怎么这么容易?
开发语言·人工智能·python·数据挖掘·数据分析·numpy
面朝大海,春不暖,花不开4 小时前
使用 Python 实现 ETL 流程:从文本文件提取到数据处理的全面指南
python·etl·原型模式
2301_805054565 小时前
Python训练营打卡Day59(2025.7.3)
开发语言·python
万千思绪6 小时前
【PyCharm 2025.1.2配置debug】
ide·python·pycharm
微风粼粼7 小时前
程序员在线接单
java·jvm·后端·python·eclipse·tomcat·dubbo
云天徽上7 小时前
【PaddleOCR】OCR表格识别数据集介绍,包含PubTabNet、好未来表格识别、WTW中文场景表格等数据,持续更新中......
python·ocr·文字识别·表格识别·paddleocr·pp-ocrv5
你怎么知道我是队长8 小时前
python-input内置函数
开发语言·python
站在巨人肩膀上的码农8 小时前
全志T507 音频ALSA核心层注册流程分析
驱动开发·音视频·安卓·全志·alsa·声卡