lite.ai.toolkit 图像分割与抠图完全指南DeepLabV3、人像分割、头发分割与 MODNet 抠像快速上手【免费下载链接】lite.ai.toolkitA lite C AI toolkit: 100 models with MNN, ORT and TRT, including Det, Seg, Stable-Diffusion, Face-Fusion.项目地址: https://gitcode.com/gh_mirrors/lit/lite.ai.toolkitlite.ai.toolkit 是一个轻量级 C AI 工具包内置 100 预置模型同时支持 MNN、ONNX Runtime、TensorRT 等主流推理后端。本文带你完整掌握其中的图像分割与抠图能力DeepLabV3 语义分割、人像分割SINet、头发分割HairSeg与 MODNet 发丝级抠像配合swap_background工具函数即可实现一键换背景 。一、分割与抠图模型一览 lite.ai.toolkit 将视觉能力统一封装为「构造 → 传入图像 → 读取结果」的极简 API分割与抠图相关模型包括能力模型源码位置输出结果语义分割DeepLabV3 ResNet10120 类 COCOlite/ort/cv/deeplabv3_resnet101.cpp彩色分割图 类别名称语义分割FCN ResNet101lite/ort/cv/fcn_resnet101.cpp彩色分割图人像分割SINet / Extremec3Netlite/ort/cv/portrait_seg_sinet.cpp人像 Mask头部分割HeadSeglite/ort/cv/head_seg.cpp头部 Mask头发分割HairSeg / MobileHairSeglite/ort/cv/hair_seg.cpp头发 Mask人像抠图MODNet / MODNet_dynlite/ort/cv/modnet.cppAlpha 透明通道人像抠图MGMatting / MobileHumanMattinglite/ort/cv/mg_matting.cppAlpha 透明通道视频抠图RVMRobust Video Mattinglite/ort/cv/rvm.cpp视频帧 Alpha各模型均提供 ORT、MNN、NCNN、TNN 等多套后端实现如 lite/mnn/cv/mnn_modnet.cpp同一业务逻辑可跨端复用。二、DeepLabV3 ResNet10120 类场景语义分割 DeepLabV3 ResNet101 是通用的语义分割模型支持 20 个 COCO 类别person、car、dog、sheep 等输入支持动态分辨率。它适合「先理解画面结构」的场景知道图中哪些像素属于人、车、动物再叠加业务逻辑。模型内部已内置 ImageNet 均值/方差归一化你只需把 OpenCV 图像直接传入detect()即可得到SegmentContent结果其中color_mat为彩色分割图、names_map为检测到的类别名称映射示例测试见 examples/lite/cv/test_lite_deeplabv3_resnet101.cpp。 提示示例中使用 16 线程推理分割结果与原图addWeighted叠加后可直观查看每个像素的类别归属。三、人像分割与头发分割精准 Mask 输出 人像类分割模型输出黑白 Mask0/255 或 0~1 浮点可直接用于裁剪、打码、美颜等后处理PortraitSegSINet / PortraitSegExtremec3Net专注人像区域分割输入 224×224自动处理缩放与反归位输出PortraitSegContent.mask支持score_threshold阈值过滤与remove_noise去噪HeadSeg针对头部肩颈区域适合头像类应用HairSeg / MobileHairSeg仅分割头发区域适合发型编辑、发色替换等业务。模型文件见 examples/hub/onnx/cv/ext_portrait_seg_SINet_224x224.onnx完整测试代码见 examples/lite/cv/test_lite_portrait_seg_sinet.cpp、examples/lite/cv/test_lite_hair_seg.cpp。人像分割的典型输出效果如下左原图右人像 Mask四、MODNet 人像抠像发丝级 Alpha 透明通道 ✨如果需要带半透明边缘的精细抠图比如发丝应选择图像抠图Matting模型。MODNet 输入 512×512输出MattingContent包含pha_matAlpha 透明通道矩阵0~1发丝处呈现灰度过渡fgr_mat前景图已乘以 Alphamerge_mat合成结果。detect()还提供remove_noise去除噪点与minimum_post_process最小后处理两个开关可一键得到干净的前景色。测试示例见 examples/lite/cv/test_lite_modnet.cpp模型文件为 examples/hub/onnx/cv/modnet_photographic_portrait_matting-512x512.onnx。五、一键换背景swap_background 工具函数 抠图之后最常见的需求是替换背景。lite.ai.toolkit 在 lite/utils.h 中提供了lite::utils::swap_background()只需传入前景图、Alpha 通道与新背景图即可完成合成另有remove_small_connected_area()用于清除 Alpha 图中的小面积噪点。整体流程MODNet.detect() → 拿到 pha_mat → swap_background() → 得到新背景照片三步完成无需自己写像素级合成逻辑。六、快速上手路径 克隆仓库git clone https://gitcode.com/gh_mirrors/lit/lite.ai.toolkit编译时按需开启ENABLE_ONNXRUNTIME/ENABLE_MNN等后端宏CMake 配置见 cmake/onnxruntime.cmake、cmake/MNN.cmake打开examples/lite/cv/下对应的分割/抠图示例替换模型路径与图片路径即可运行各接口的详细说明可参考官方文档 docs/api/api.default.md。七、总结 lite.ai.toolkit 把图像分割与抠图做成了「开箱即用」的组件DeepLabV3 负责通用场景理解SINet/HeadSeg/HairSeg 输出各类精准 MaskMODNet 则提供发丝级 Alpha 抠像配合swap_background工具函数从理解画面到更换背景只需几行代码。无论面向移动端MNN/NCNN还是服务端ONNX Runtime/TensorRT同一套 API 都能无缝复用。【免费下载链接】lite.ai.toolkitA lite C AI toolkit: 100 models with MNN, ORT and TRT, including Det, Seg, Stable-Diffusion, Face-Fusion.项目地址: https://gitcode.com/gh_mirrors/lit/lite.ai.toolkit创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考