jnulzl/cv_unet_image-matting

BSHM人像抠图ONNX版本

BSHM人像抠图ONNX版本

原始模型见：modelscope-cv_unet_image-matting

安装依赖

>>python --version
Python 3.9.13
>>pip install -r requirements.txt

demo

• 1、直接输出抠完的图像

python onnx_demo_matting_output_png.py --model sim/cv_unet_image_matting_opset11_output_png.onnx --src_img_path imgs/1.png

原图：

人像扣图：

• 2、只输出人像mask

python onnx_demo_matting_output_mask.py --model sim/cv_unet_image_matting_opset11_output_float_mask.onnx --src_img_path imgs/1.png

原图：

人像mask：

一些有用的onnx模型处理脚本