Skip to content

jnulzl/cv_unet_image-matting

Repository files navigation

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

原图:

wx

人像扣图:

wx.jpgmerge

  • 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

原图:

wx

人像mask

wx.jpg_mask

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

Releases

No releases published

Packages

No packages published

Languages