scatterco / automatteprocessor Goto Github PK

my env is below
running on windows 10
gtx 750 it 2G(video ram)

nVidia CUDA runtime 11.0
nVidia GPU dirver ver 456.71

Python 3.9.5 (tags/v3.9.5:0a7dcbd, May 3 2021, 17:27:52) [MSC v.1928 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.

C:\Users\Administrator>pip list
Package Version

---

absl-py 0.12.0
ansicolors 1.1.8
ansiwrap 0.8.4
backcall 0.2.0
cachetools 4.2.2
certifi 2020.12.5
chainmap 1.0.3
chardet 4.0.0
cmake 3.18.4.post1
colorama 0.4.4
combomethod 1.0.12
cycler 0.10.0
decorator 5.0.7
dlib 19.22.0
google-auth 1.30.0
google-auth-oauthlib 0.4.4
grpcio 1.37.1
idna 2.10
intspan 1.6.1
ipython-genutils 0.2.0
jedi 0.18.0
kiwisolver 1.3.1
kornia 0.4.1
Markdown 3.3.4
matplotlib 3.4.2
matplotlib-inline 0.1.2
mementos 1.3.1
MouseInfo 0.1.3
nulltype 2.3.1
numpy 1.20.3
oauthlib 3.1.0
onnxruntime 1.7.0
opencv-python 4.4.0.46
options 1.4.10
parso 0.8.2
pickleshare 0.7.5
Pillow 8.2.0
pip 21.1.1
prompt-toolkit 3.0.18
protobuf 3.17.0
pyasn1 0.4.8
pyasn1-modules 0.2.8
PyAutoGUI 0.9.52
PyGetWindow 0.0.9
Pygments 2.9.0
PyMsgBox 1.0.9
pyparsing 2.4.7
pyperclip 1.8.2
PyRect 0.1.4
PyScreeze 0.1.27
python-dateutil 2.8.1
PyTweening 1.0.3
requests 2.25.1
requests-oauthlib 1.3.0
rsa 4.7.2
say 1.6.6
setuptools 56.0.0
simplere 1.2.13
six 1.12.0
tensorboard 2.3.0
tensorboard-plugin-wit 1.8.0
textdata 2.4.1
textwrap3 0.9.2
torch 1.7.1+cu110
torchvision 0.8.2+cu110
tqdm 4.51.0
traitlets 5.0.5
typing-extensions 3.10.0.0
urllib3 1.26.4
wcwidth 0.2.5
Werkzeug 2.0.0
wheel 0.36.2

=============================================

E:\pybin\auto\AutoMatteProcessor>python ProcessDirectory.py movie
E:\pybin\auto\AutoMatteProcessor\BackgroundMattingV2\inference_video.py
Models\PyTorch\pytorch_resnet50.pth
Processing directory: movie
Found video file 20.mp4
Outputting to temporary dir: movie__tmp\20
Running inference script
100%|████████████████████████████████████████████████████████████████████████████████| 336/336 [00:35<00:00, 9.43it/s]
Copying output file from movie__tmp\20\pha.mp4 to movie\20_matte.mp4
Cleaning up temporary output directory: movie__tmp\20
Done processing
Cleaning temporary output directory: movie__tmp

without error reporting , but get a all-white-video.(mp4 file)

looking forward to your answer