[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文件代码进行修改

相关推荐
Warren9833 分钟前
Java Collections工具类
java·开发语言·笔记·python·学习·oracle·硬件工程
love530love38 分钟前
Windows 11 下 Anaconda 命令修复指南及常见问题解决
运维·ide·人工智能·windows·python·架构·conda
NeoFii39 分钟前
Day 24:元组与os模块
python·机器学习
半新半旧1 小时前
1.DRF 环境安装与配置
python·django
Sean_summer1 小时前
暑期第二周
前端·数据库·python
封奚泽优1 小时前
使用Python绘制金融数据可视化工具
python·信息可视化·excel·pandas·pyplot·qtwidgets·qtcore
数据智能老司机1 小时前
使用 Python 进行并行与高性能编程——构建多线程程序
python·性能优化·编程语言
站大爷IP1 小时前
Python极速搭建局域网文件共享服务器:一行命令实现HTTPS安全传输
python
执器1 小时前
SIFT实现简单的图像匹配,查找icon在图中的位置
python·图像识别
Ronin-Lotus1 小时前
程序代码篇---数据包解析
python·esp32·数据包解析