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

git clone code
cd yolov11-gradcam

conda create -n yolov11-gradcam python=3.8
conda activate yolov11-gradcam

pip install torch==2.1.1+cu118 torchvision==0.16.1+cu118 -f https://download.pytorch.org/whl/torch_stable.html
pip install -r requirements.txt

from Yolov11Heatmap import *

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

from Yolov11Heatmap import *

model = Yolov11Heatmap(weight='yolo11n.pt')
model.start_video('test.mp4')

from Yolov11Heatmap import *

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