0

0

Python实现机器学习中目标检测的详细教程【教程】

舞夢輝影

舞夢輝影

发布时间:2025-12-18 13:31:02

|

539人浏览过

|

来源于php中文网

原创

目标检测需同时识别物体类别和定位边界框;推荐用YOLOv8(Ultralytics)+自定义数据集入门,按YOLO格式组织数据、用LabelImg标注、编写data.yaml配置、一行命令训练、三行代码推理并可视化结果。

python实现机器学习中目标检测的详细教程【教程】

目标检测不是分类也不是单纯定位,而是既要识别图像中有哪些物体(类别),又要标出它们的位置(边界框)。Python生态里最实用的入门路径是用 YOLOv8(Ultralytics 实现)+ 自定义数据集,不依赖复杂环境配置,代码简洁、训练快、效果稳。

准备数据:按YOLO格式组织图像和标注

YOLO要求每张图对应一个同名 .txt 文件,每行描述一个目标:类别索引 x_center y_center width height(归一化到 0~1)。比如一张 640×480 图片里有个猫,框左上(100,80)、宽200、高150,则中心点为 (200,155),归一化后是 0 0.3125 0.3229 0.3125 0.3125(类别0=cat)。

建议操作:

  • LabelImgCVAT 标注图像,导出为 YOLO 格式
  • 目录结构统一为:dataset/images/train/dataset/labels/train/dataset/images/val/dataset/labels/val/
  • 写个简单脚本检查 .txt 是否缺失、坐标是否越界(x,y,w,h 都应在 [0,1] 内)

安装与加载模型:一行命令启动训练

Ultralytics 的 YOLOv8 封装极好,无需手动写 Dataloader 或 Loss 函数。

立即学习Python免费学习笔记(深入)”;

执行:

pip install ultralytics

然后直接调用预训练模型开始微调:

析稿Ai写作
析稿Ai写作

科研人的高效工具:AI论文自动生成,十分钟万字,无限大纲规划写作思路。

下载
from ultralytics import YOLO

加载预训练权重(自动下载)

model = YOLO('yolov8n.pt') # nano 版,轻量适合入门

训练(指定数据配置文件 + 超参)

model.train( data='dataset/data.yaml', # 包含 train/val 路径和 nc/classes 的 YAML epochs=50, imgsz=640, batch=16, name='my_cat_detector' )

data.yaml 示例:

train: ../dataset/images/train
val: ../dataset/images/val

nc: 1 names: ['cat']

推理与可视化:三行代码跑通检测流程

训练完模型会保存在 runs/detect/my_cat_detector/weights/best.pt,直接加载即可预测:

model = YOLO('runs/detect/my_cat_detector/weights/best.pt')
results = model('test.jpg')  # 返回 Results 对象
results[0].show()  # 弹窗显示带框结果

若需提取结果:

  • boxes = results[0].boxes → 获取 xyxy 坐标、置信度、类别
  • cls = boxes.cls.tolist()conf = boxes.conf.tolist()xyxy = boxes.xyxy.tolist()
  • 用 OpenCV 或 matplotlib 自定义画框、加标签、保存图像

常见问题快速排查

训练不动?检测全漏?先看这几点:

  • 标注文件名必须和图片完全一致(包括大小写、空格、扩展名),且放在对应 labels/ 目录下
  • YAML 中的 nc(类别数)和 names 必须与你的数据严格匹配
  • 验证集 loss 不下降?检查 val 图像是否混入了 train 标注,或存在未标注的图
  • 预测框偏移?确认训练时 imgsz 和推理时输入尺寸一致,YOLO 默认会 resize 保持长宽比

基本上就这些。YOLOv8 把工程细节封装得很干净,真正需要你动脑的是数据质量和问题定义——框标得准不准,类别分得清不清,比调参重要得多。

相关专题

更多
python开发工具
python开发工具

php中文网为大家提供各种python开发工具,好的开发工具,可帮助开发者攻克编程学习中的基础障碍,理解每一行源代码在程序执行时在计算机中的过程。php中文网还为大家带来python相关课程以及相关文章等内容,供大家免费下载使用。

765

2023.06.15

python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

639

2023.07.20

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

764

2023.07.25

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

619

2023.07.31

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

1285

2023.08.03

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

549

2023.08.04

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

579

2023.08.04

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

709

2023.08.11

PS使用蒙版相关教程
PS使用蒙版相关教程

本专题整合了ps使用蒙版相关教程,阅读专题下面的文章了解更多详细内容。

23

2026.01.19

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 5.4万人学习

Django 教程
Django 教程

共28课时 | 3.3万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.2万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号