PeterL1n / RobustVideoMatting

Robust Video Matting in PyTorch, TensorFlow, TensorFlow.js, ONNX, CoreML!
https://peterl1n.github.io/RobustVideoMatting/
GNU General Public License v3.0
8.61k stars 1.14k forks source link

Background. #157

Closed DineshJoshi2 closed 2 years ago

DineshJoshi2 commented 2 years ago

Is it possible to have a white background instead of green ?

surifans commented 2 years ago

bgr = torch.tensor([120, 255, 155], device=device, dtype=dtype).div(255).view(1, 1, 3, 1, 1)

把[120, 255, 155]改成[255, 255, 255]即可