这是很多电商平台或图片工具网站的付费功能,以1688上面的一个飞机模型玩具为例用代码进行抠图。
模型
抠图的时候,通过model属性可以根据不同的应用场景选择不同的处理模型。支持的模型如下:
- u2net - 默认模型,通用性最好
- u2netp - u2net的轻量级版本
- u2net_human_seg - 专门用于人像分割
- silueta - 小型高效模型
- isnet-general-use - 通用场景模型
- isnet-anime - 动漫图像专用
- sam - 基于Segment Anything Model
执行前需安装rembg和onnxruntime库,第一次执行会下载u2net.onnx到本地。
from rembg import remove
from PIL import Image, ImageEnhance
import cv2
import<