4d3fbe899bbc574caa3daf6b46c5ec36533b934e
图像二值化处理
运行库
cv2、matplotlib、PIL、rembg
onnxruntime(rembg需要用到此库)
rembg库
这个库可以对图像中的主体进行抠图处理,去掉背景仅保留人物/动物等。
初次运行代码会自动下载v2net.onnx模型文件(必须),模型文件大小为176M。安装这个模型需要进行魔法上网。
rembg仓库地址
二值化处理
_, binary_image = cv2.threshold(image, 1, 255, cv2.THRESH_BINARY)
将图像灰度值 > 1的像素变为255(白色),也就是将抠图后保留的部分完全设为白色。其他区域像素全为0(黑色)。
结果展示
致谢
Description
利用Python的rembg库实现对图像主体抠图,再利用二值化将图片主题区域转化为白色背景为黑色。
1.2 MiB
0 Stars
1 Watchers
0 Forks
Languages
Python
100%
