andylucny / BackgroundRemovalByModNet4OpenCV

ModNet for pure OpenCV
20 stars 7 forks source link

BackgroundRemovalByModNet4OpenCV

Background Removal by ModNet in the pure OpenCV

input image output image
bgremoval-1 bgremoval-2

This project reproduces two models for background removal distributed at https://github.com/ZHKKKe/MODNet under Creative Commons Attribution NonCommercial ShareAlike 4.0 license, i.e. for non-commercial purposes only

The original models are in the pytorch .ckpt format and cannot be loaded to OpenCV. Here they are offered in .onnx format (version 9) with a demo source code, tested with OpenCV 4.5.

Download onnx model for the web camera

Download onnx model for photos

The demo source code is free.