Efficient-Computing icon indicating copy to clipboard operation
Efficient-Computing copied to clipboard

about video inference

Open QiqLiang opened this issue 1 year ago • 1 comments

你好,我想用训练好的模型在自己本地的.mp4视频文件上做推理,推理可视化结果保存为.mp4文件,请问我要怎么修改代码呢?谢谢!

QiqLiang avatar Oct 25 '23 09:10 QiqLiang

我认为你应该翻译训练后获得的权重weights.pt 到ONNX模型,然后将其转换到TensorRT引擎。 然后在网上找到或者自己写一个c++中的推理代码,将使用TensorRT引擎。

Egorundel avatar Oct 25 '23 13:10 Egorundel