在本文中,我们将展示如何使用 YOLOv10 模型对图片进行目标检测,并将检测结果保存为标签文件。生成的标签文件采用 YOLO 格式,仅包含类别编号和归一化的目标框信息(中心坐标、宽度和高度),不包含置信度。
此外,通过 project
参数,可以方便地指定检测结果的保存路径。
一、背景介绍
YOLOv10 是一款高效的目标检测模型,其 predict()
方法可直接对输入图片或视频进行推理。默认情况下,该方法返回一个 Results
列表,每个 Results
对象中封装了检测框、类别、置信度等信息。但该方法不会自动生成标签文件,本文将展示如何将检测结果保存为 .txt
文件。
二、完整代码示例
from ultralytics import YOLOv10
import os
# =======&