讲师中心 微信公众号
AI工具推荐 视频效率加速

Python深度学习如何训练多标签图像分类模型的关键细节【教程】

轻伟小哥_6911

轻伟小哥_6911

发布时间:2025-12-15 20:38:02

|

545人浏览过

|

来源于php中文网

原创

多标签图像分类需用二值向量标签、binary_crossentropy损失、sigmoid输出及多标签专用评估指标。标签为shape=(batch_size, num_classes)的0/1张量,损失函数须为binary_crossentropy且输出层用sigmoid,评估宜用Hamming loss、subset accuracy和macro-F1。

python深度学习如何训练多标签图像分类模型的关键细节【教程】

多标签图像分类和单标签不同,一张图可能同时属于多个类别(比如“猫”“室内”“白天”),模型输出不再是单一概率分布,而是每个标签独立的置信度。关键不在换模型,而在数据准备、损失函数、评估方式和预测逻辑这四点上。

标签格式必须是二值向量,不是整数索引

单标签用 [0, 0, 1, 0] 表示第3类;多标签则用 [1, 0, 1, 1] 表示同时属于第0、第2、第3类。输入模型前,标签要转成 shape=(batch_size, num_classes) 的 float32 张量,值只能是 0 或 1。

  • 用 sklearn.preprocessing.MultiLabelBinarizer 处理原始标签列表(如 [['cat','indoor'], ['dog','outdoor']])最稳妥
  • 别用 to_categorical —— 它只适合单标签,会把多标签错误展开成稀疏独热码
  • 验证时打印几个样本的 label.sum(axis=1),确保结果可以大于 1(说明允许多个 1)

损失函数要用 binary_crossentropy,不是 categorical_crossentropy

因为每个标签是独立的二分类任务,不是互斥的多选一。Keras 中直接指定 loss='binary_crossentropy' 即可,但要注意:

  • 最后一层激活必须是 sigmoid(输出 0~1 概率),不能用 softmax
  • 如果手动写 loss,确保用 tf.keras.losses.BinaryCrossentropy(from_logits=False)
  • 别在标签里做归一化或除以总和——那会破坏二值语义

预测后需设定统一阈值,再按需调整

模型输出是每个类的概率值(如 [0.82, 0.15, 0.93, 0.67]),需转成 0/1 才能算准确率。常用做法是:

Conventional Git
Conventional Git

Conventional Commits v1.0.0 分支、工作树命名及提交信息规范,适用于 GitHub 与 GitLab 项目,用于创建分支和命名工作树等场景。

下载

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

  • 初始统一用 threshold=0.5:大于即判定存在
  • 若某类样本极少(如“金毛犬”只占 0.3%),可单独调低该类阈值(用 per-class threshold)
  • 更稳的方式是用 validation set 跑 grid search,在 F1-score 或 hamming loss 上找最优全局阈值

评估指标不能只看 accuracy,得用多标签专用指标

accuracy 在多标签中意义不大(全错一个也算整体错),推荐组合使用:

  • Hamming loss:错误标签占总标签数的比例,越低越好(sklearn.metrics.hamming_loss)
  • Exact match ratio (subset accuracy):整张图所有标签全对才算对,非常严格(sklearn.metrics.accuracy_score,需加 normalize=True)
  • Per-class F1:分别算每个类的 precision/recall/F1,再 macro 平均,能看出长尾类表现

基本上就这些。不复杂但容易忽略——尤其标签格式和损失函数配错,模型会训得“看起来收敛”实则学不会多标签逻辑。

热门AI工具

更多
Seko
Seko Hot

一款AI视频创作工具,主要用于商汤科技推出的创编一体的AI短视频创作Agent,适合需要提升相关任务效率的用户。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

切问学术

切问学术是一款AI论文写作工具,复旦大学NLP团队推出的AI学术智能体。

UpDream
UpDream Hot

一款AI视频创作工具,主要用于哔哩哔哩推出的自研AI视频创作工具,适合需要提升相关任务效率的用户。

Laper
Laper Hot

Laper是专为编剧、导演和制片人推出的 AI 原生剧本创作工具。

AionClaw
AionClaw Hot

AionClaw是一款面向办公、创作和编程任务的AI桌面智能体。

超级简历WonderCV

一款AI办公效率工具,主要用于免费求职简历模版下载制作,应届生职场人必备简历制作神器,适合需要提升相关任务效率的用户。

相关专题

更多
class在c语言中的意思
class在c语言中的意思

在C语言中,"class" 是一个关键字,用于定义一个类。想了解更多class的相关内容,可以阅读本专题下面的文章。

3216

2024.01.03

python中class的含义
python中class的含义

本专题整合了python中class的相关内容,阅读专题下面的文章了解更多详细内容。

286

2025.12.06

Buffalo框架数据库开发全教程
Buffalo框架数据库开发全教程

本专题围绕Buffalo框架数据库开发,讲解database.yml多环境配置、soda与fizz迁移生成回滚、模型结构体标签、增删改查与条件查询、一对多与多对多关联、数据校验、回调钩子、事务处理及原生SQL执行能力。

80

2026.09.23

Buffalo框架路由与请求处理实操指南
Buffalo框架路由与请求处理实操指南

本专题讲解Buffalo框架路由与请求处理机制,涵盖路由注册与分组、资源路由、Handler编写规范、Context上下文方法、参数绑定、中间件编写挂载、Session与Cookie读写、Flash消息及错误页面定制方法。

20

2026.09.23

Buffalo框架零基础入门教程
Buffalo框架零基础入门教程

本专题整理Buffalo框架入门内容,涵盖Go环境准备、buffalo CLI安装、新项目生成、目录结构说明、dev热加载启动、数据库连接配置与常见报错排查,帮助新手按约定优于配置的思路跑通第一个Buffalo框架应用。

20

2026.09.23

Conan创建软件包配方指南
Conan创建软件包配方指南

本专题介绍通过conanfile.py创建软件包的方法,讲解包名、版本、依赖和构建设置等基础信息,以及source、build、package、package_info等常用方法的作用及编写思路。

20

2026.09.22

Conan二进制包配置指南
Conan二进制包配置指南

本专题介绍Conan根据操作系统、编译器、架构和构建类型生成二进制包的方法,讲解Profile、Settings、Options及Package ID的作用,帮助管理不同平台和编译环境下的包版本。

40

2026.09.22

Conan私有仓库搭建教程
Conan私有仓库搭建教程

本专题系统的讲解Conan私有仓库的搭建流程,涵盖仓库服务部署、存储目录配置、用户认证、权限划分和远程地址添加,并介绍内部C++依赖包的上传、下载及版本维护方法。

20

2026.09.22

loomy官网入口地址合集
loomy官网入口地址合集

本专题汇总了 Loomy 桌面 AI 助理的官方入口地址合集及使用指南。提供 macOS 与 Windows 客户端下载 。Loomy 是讯飞推出的桌面级 AI 工作搭子,支持文件整理、数据分析、网页操作及通过飞书/钉钉远程操控电脑,助你高效完成本地办公任务 。

20

2026.09.22

热门下载

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

精品课程

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

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn