RVM_onnx_compose

RobustVideoMatting and background composing in one model by using onnxruntime.

Usage

pip install -r requirements.txt
python infer_cam_onnx_freeze.py

Press key 1~4 for changing 4 background images in folder backgrounds, press key 5 for using video as background, press 0 for green background, press q for raw web-cam stream.

To do

Solve memory leak, which may caused by cap.read()