search

PeterL1n / RobustVideoMatting

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

推理速度是2~3帧每秒 #159

Closed napohou closed 2 years ago

napohou commented 2 years ago

V100的GPU，用的模型是预训练的mobilenetv3.pth，视频是1920*1080的，请问是不是太慢？

napohou commented 2 years ago

GPU占用很低，经常在3%左右

PeterL1n commented 2 years ago

inference.py 的 video encoding 和 decoding 不是 hardware accelerate 的。你要自己写硬件加速的推理