讲师中心 微信公众号

工具推荐

视频效率加速
AniShort
AniShort

AI短剧协同制作平台

立刻MV
立刻MV

低成本歌曲 MV、视频生成

updream

一键生成高质量推广成片

Laper
Laper

剧本策划到成片协作

如何在 MediaPipe Model Maker 中自定义图像增强类型

胖浩姑娘_3628

胖浩姑娘_3628

发布时间:2026-01-05 15:05:29

|

780人浏览过

|

来源于php中文网

原创

如何在 MediaPipe Model Maker 中自定义图像增强类型

截至 mediapipe model maker v0.10.9,官方不支持直接配置特定增强操作(如仅保留水平翻转、曝光调整和模糊),其 `do_data_augmentation` 仅为开关式布尔参数,内置增强逻辑固定且不可定制。

MediaPipe Model Maker 的 image_classifier 模块为快速部署提供了便捷封装,但在数据增强(Data Augmentation)层面高度抽象化。当前版本中,HParams(do_data_augmentation=True/False) 仅控制是否启用预设的默认增强流水线,该流水线由底层 ImagePreprocessor 实现,包含且仅包含两项操作:

  • ✅ 随机水平翻转(tf.image.random_flip_left_right)
  • ✅ 随机扭曲裁剪(tf.image.sample_distorted_bounding_box,会破坏原始宽高比,后续缩放可能导致形变)

⚠️ 重要限制说明:

  • ❌ 不支持单独启用/禁用某类增强(例如“只开翻转、关裁剪”);
  • ❌ 官方未实现曝光(exposure)、高斯模糊(blur)、色彩抖动、对比度调整等常见增强;
  • ❌ 无公开 API 或配置项允许传入自定义 tf.keras.layers 或 tf.image 函数链。

替代方案(推荐实践)

若需精确控制增强策略(如仅应用 random_flip_left_right + adjust_exposure + gaussian_blur),有以下两种可行路径:

Python Hardener
Python Hardener

通过完整的安全审查、代码整洁重构、健壮的错误处理、正确的日志记录以及完整的文档字符串覆盖,强化 Python 脚本。

下载

✅ 方案一:预处理数据集(推荐,简洁安全)

在调用 ImageClassifier.create() 前,对原始训练数据进行离线增强,并将结果保存为新数据集:

import tensorflow as tf
import numpy as np

def custom_augment(image, label):
    # 1. 随机水平翻转
    image = tf.image.random_flip_left_right(image)
    # 2. 调整曝光(模拟 gamma 校正)
    image = tf.image.adjust_gamma(image, gamma=0.8 + tf.random.uniform([]) * 0.4)
    # 3. 添加轻微高斯模糊(需自定义核)
    blur_kernel = tf.constant([
        [1, 2, 1],
        [2, 4, 2],
        [1, 2, 1]
    ], dtype=tf.float32) / 16.0
    blur_kernel = blur_kernel[:, :, tf.newaxis, tf.newaxis]
    image = tf.nn.conv2d(
        tf.expand_dims(image, 0), 
        blur_kernel, 
        strides=1, 
        padding='SAME'
    )[0]
    return tf.clip_by_value(image, 0.0, 255.0), label

# 应用于 tf.data.Dataset
train_data = train_data.map(custom_augment, num_parallel_calls=tf.data.AUTOTUNE)
train_data = train_data.cache().shuffle(1000).batch(32).prefetch(tf.data.AUTOTUNE)

# 创建模型时关闭内置增强
hparams = image_classifier.HParams(
    epochs=100,
    export_dir="exported_model_2",
    do_data_augmentation=False  # ← 关键:避免重复增强
)

⚙️ 方案二:继承并重写核心类(高级,需维护)

通过继承 ImageClassifier 并覆盖 _create_preprocessing_layer() 或修改 ImagePreprocessor 行为实现深度定制。但需注意:

  • 需 fork MediaPipe 仓库并修改源码(如 mediapipe/model_maker/python/vision/core/image_preprocessing.py);
  • 后续升级 Model Maker 时需手动同步变更,维护成本高;
  • 不适用于生产环境快速迭代场景。

总结建议

对于绝大多数项目,预处理数据集(方案一)是更鲁棒、可复现且符合工程最佳实践的选择。它完全绕过框架限制,赋予你对每一步增强的绝对控制权,同时保持与 MediaPipe Model Maker 训练流程的兼容性。务必在增强后关闭 do_data_augmentation,防止双重增强导致分布偏移。未来若 MediaPipe 开放 augmentation_fn 参数或 CustomAugmentationOptions,可再平滑迁移。

热门AI工具

更多
豆包大模型

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

函子Zion

一款支持AI搭建、可视化编辑和多端发布的全栈无代码开发平台。

WorkBuddy

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

讯飞绘文

讯飞绘文是一款由科大讯飞推出的一站式 AIGC 内容运营平台。

Seko
Seko Hot

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

VibeKnow
VibeKnow Hot

一款AI视频创作工具,主要用于全球首个AI知识视频创作平台,文档、文章、网页,一键生成视频,适合需要提升相关任务效率的用户。

PixPix
PixPix Hot

PixPix是一款面向电商视觉生产的AI商品图生成工具。

DeepSeek

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

音述AI
音述AI Hot

一款AI音频处理工具,主要用于音述AI是一个以“用声音述说故事”为核心的 AI 音乐创作与声音分享社区,适合需要提升相关任务效率的用户。

相关专题

更多
Go-Zero API开发使用教程
Go-Zero API开发使用教程

本专题聚焦Go-Zero的API开发,拆解.api描述文件的语法结构、路由与分组定义、参数绑定与校验规则,说明统一响应体、JWT鉴权、跨域与接口文档生成的落地方式,并给出网关聚合与接口测试的实践建议。

0

2026.11.12

Go-Zero安装与goctl环境配置教程
Go-Zero安装与goctl环境配置教程

本专题围绕Go-Zero的安装与工具链配置展开,讲清Go版本要求、goctl安装方式、代理镜像、protoc依赖与环境变量设置,说明多版本共存、升级回退与CI流水线安装的做法,并整理安装阶段的常见报错与验证步骤。

0

2026.11.12

Go-Zero零基础入门教程
Go-Zero零基础入门教程

本专题整理Go-Zero的入门知识,从框架定位、环境准备到跑通第一个API服务,讲清目录分层、配置加载、日志与中间件的基本用法,并对比Gin、Kratos的差异,帮助Go开发者建立整体认知,判断它是否适合当前项目。

0

2026.11.12

Postman身份认证怎么配置
Postman身份认证怎么配置

本专题从实际API接口调试场景出发,讲解Postman身份认证的完整配置流程,包括常见认证协议介绍、授权参数设置、访问令牌填写、OAuth 2.0授权及认证失败排查,帮助用户正确完成接口授权与访问验证。

120

2026.11.06

Postman请求头Headers怎么设置
Postman请求头Headers怎么设置

围绕Postman HTTP请求头配置展开,讲解Headers选项的具体使用方式,包括Key和Value参数填写、请求头信息添加、常见Header字段设置及请求头修改方法,帮助用户解决接口调试过程中遇到的请求头配置问题。

60

2026.11.06

Postman接口请求发送教程
Postman接口请求发送教程

详细介绍Postman接口请求的创建与发送方法,涵盖GET和POST请求、URL配置、请求参数、Headers请求头、Body请求体设置以及响应结果查看,帮助用户掌握Postman接口请求的完整操作流程。

80

2026.11.06

DBeaver怎么导入数据
DBeaver怎么导入数据

DBeaver数据导入专题,整理Excel、CSV、SQL、TXT等常见文件的数据导入方法,涵盖目标数据库选择、数据表创建、字段映射、数据类型设置及导入参数配置等操作,帮助用户将本地文件中的数据写入数据库,并了解批量导入、重复数据处理和导入失败等常见问题的解决方法。

80

2026.11.05

DBeaver安全防护教程大全
DBeaver安全防护教程大全

DBeaver安全防护专题,围绕数据库连接安全、账号密码保护和访问权限管理展开,整理SSL/TLS加密连接、SSH隧道、数据库凭据保存及证书验证等配置方法,同时介绍敏感信息保护、远程连接风险和高权限账号使用等注意事项,帮助用户提高数据库日常管理的安全性。

60

2026.11.05

DBeaver连接创建与配置教程
DBeaver连接创建与配置教程

本专题聚焦DBeaver数据库连接配置与管理,覆盖从新建连接、选择数据库类型到填写连接参数和验证连接的完整流程。针对MySQL、PostgreSQL、SQL Server等常见数据库,整理相应的连接设置方法,并提供驱动缺失、认证失败和连接异常等问题的排查方向。

60

2026.11.04

热门下载

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

精品课程

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

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