YOLOV8特征图可视化

YOLOV8进行图片/视频的预测并输出特征图

可以使用自己创建的predict.py文件

from ultralytics import YOLO

#model = YOLO("yolov8n.pt")
model = YOLO("D:/yolov8/ultralytics-main/ultralytics-main/runs/detect/train90/weights/best.pt")  # 模型文件路径

results = model("D:/yolov8/1.png", visualize=True)  # 要预测图片路径和使用可视化
1
2
3
4
5
6

或者直接调用YOLOV8的预测命令，关键是加入visualize=True的命令
运行后会保存每层预测图片/视频的特征图