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()