图像二值化处理

运行库

cv2、matplotlib、PIL、rembg
onnxruntime(rembg需要用到此库)

rembg库

这个库可以对图像中的主体进行抠图处理，去掉背景仅保留人物/动物等。
初次运行代码会自动下载v2net.onnx模型文件(必须)，模型文件大小为176M。安装这个模型需要进行魔法上网。
rembg仓库地址

二值化处理

_, binary_image = cv2.threshold(image, 1, 255, cv2.THRESH_BINARY)

将图像灰度值 > 1的像素变为255(白色)，也就是将抠图后保留的部分完全设为白色。其他区域像素全为0(黑色)。

结果展示

原图、抠图后的图片、二值化处理抠图后的图片的图片。

致谢

感谢作者Bill FANG的文章《采用Python的rembg库去除图片背景》
感谢rembg项目的每一个成员。