news 2026/8/6 17:21:35

FeyNoBg抠图笔记 NoBg使用示例

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
FeyNoBg抠图笔记 NoBg使用示例

FeyNoBg 是一个开源的、基于 AI 的图像背景移除模型,它能实现发丝级的高精度抠图。它的使用方法主要有两种:一是通过其背后的NoBg Python 库进行编程调用,二是通过Hugging Face Space在线体验。

由于目前公开信息主要聚焦于其库的用法,以下核心操作步骤是基于NoBg 库的 Python 工作流整理的。

使用步骤

1. 环境准备


首先,确保你的 Python 版本 >= 3.10,并安装好 PyTorch (>= 2.0)。然后通过 pip 安装 NoBg 库:

pip install nobg torch

2.抠图代码示例

被扣的是input.jpg,扣好输出是output.png

import os # 1. 关键步骤:设置环境变量,指向国内镜像源 os.environ["HF_ENDPOINT"] = "https://hf-mirror.com" import torch from loadimg import load_img from nobg import AutoModel, AutoProcessor # 2. 加载模型 (现在会从 hf-mirror.com 下载) model = AutoModel.from_pretrained("feyninc/FeyNobg").eval().to("cuda") processor = AutoProcessor.from_pretrained("feyninc/FeyNobg") # 3. 读取图片并预处理 image = load_img("input.jpg").convert("RGB") inputs = processor(image, return_tensors="pt").to("cuda") # 4. 模型推理 with torch.no_grad(), torch.autocast("cuda", dtype=torch.bfloat16): outputs = model(pixel_values=inputs["pixel_values"]) # 5. 生成透明背景图片 alpha = processor.post_process_alpha_matting( outputs, target_sizes=[(image.height, image.width)] )[0] processor.cutout(image, alpha).save("output.png")

注意

a、第一次使用,会下载模型到C:\Users\你的用户名\.cache\huggingface\hub\

models--feyninc--FeyNobg命名的文件夹,这就是 FeyNoBg 模型的存放位置。

b、因为是国内,添加了国内镜像源

c、这个是有cuda的显卡的代码

如果没cuda显卡应该这么写

import os # 1. 关键步骤:设置环境变量,指向国内镜像源 os.environ["HF_ENDPOINT"] = "https://hf-mirror.com" import torch from loadimg import load_img from nobg import AutoModel, AutoProcessor # 1. 从 Hugging Face 加载模型和处理器 model = AutoModel.from_pretrained("feyninc/FeyNobg").eval() processor = AutoProcessor.from_pretrained("feyninc/FeyNobg") # 2. 读取并预处理图片 image = load_img("input.jpg").convert("RGB") inputs = processor(image, return_tensors="pt") # 3. 模型推理(不计算梯度) with torch.no_grad(): outputs = model(pixel_values=inputs["pixel_values"]) # 4. 后处理,生成与原始图片尺寸一致的 Alpha 遮罩 alpha = processor.post_process_alpha_matting( outputs, target_sizes=[(image.height, image.width)] )[0] # 5. 合成并保存透明背景的 PNG 图片 processor.cutout(image, alpha).save("output.png")

测试成功的shell截图(含有第一次下载模型)

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/8/6 17:12:05

银河麒麟桌面系统WPS客户端清理缓存

步骤操作下 ① 确认WPS中「文件-备份与恢复-备份管理-查看其他备份」下的文件均为非必要文件,若有重要文件,请另存为至其他路径保存,此操作会清空本地备份文件 ② 下载脚本文件 ③ 打开命令终端右键-命令终端,输入chmod x&#xf…

作者头像 李华
网站建设 2026/8/6 17:11:33

【金仓数据库征文】金仓数据库高可用集群实战:读写分离、故障切换与运维管理

【金仓数据库征文】金仓数据库高可用集群实战:读写分离、故障切换与运维管理 前言 高可用是生产环境数据库系统的核心要求。金仓数据库提供了完善的高可用集群方案,包括主备复制、读写分离、自动故障切换等功能,确保业务的连续性。 本篇内容深…

作者头像 李华
网站建设 2026/8/6 17:10:14

电商购物车测试:核心检查点与性能优化策略

1. 电商购物车测试的核心价值与挑战购物车作为电商平台的核心转化枢纽,其稳定性直接影响着平台的GMV和用户体验。根据我过去五年参与7个大型电商项目的测试经验,购物车模块的缺陷平均占电商系统总缺陷数的23%,其中价格计算错误和库存同步问题…

作者头像 李华
网站建设 2026/8/6 17:06:31

TiDB与CockroachDB深度对比:NewSQL分布式数据库选型与实战指南

1. 从NewSQL到分布式数据库:为什么我们需要重新思考数据架构十年前,当我们的业务数据量开始突破单台服务器的物理极限时,我们面临的选择通常是痛苦的:要么花大价钱购买更昂贵、更大型的“小型机”和高端存储,要么就得走…

作者头像 李华