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

Error on images #61

Closed filthy11111 closed 3 years ago

filthy11111 commented 3 years ago

Hi

Running the code perfectly on videos, but when I try run the below on images i get an error.

Code: python inference.py --variant mobilenetv3 --checkpoint ./model/PyTorch/rvm_mobilenetv3.pth --device cuda --input-source ./inputimg --output-type png_sequence --output-composition ./outputimg --output-video-mbps None --seq-chunk 1

Error: inference.py: error: argument --output-video-mbps: invalid int value: 'None'

Am i missing something? running on Windows via Anacoda

Thanks!

PeterL1n commented 3 years ago

Fixed the inference.py. Try pull it.

Also, just leave out the --output-video-mbps argument when inference png_sequence output.

filthy11111 commented 3 years ago

Thanks works perfectly now.