DVQA
DVQA copied to clipboard
为什么对视频的评分需要的内存如此之大!
我需要测试的视频是yuv大小大约是433M 1366x768 大约280多帧。使用CPU机器进行评分。机器内存4核32G。 运行一段时间。提示需要52G的内存。为何需要如此之大的内存?能否解答下。或者是否有参数可以调节。
@304883563 请问您有LIVE或其他数据集可以分享下吗[email protected],非常感谢!!!
我需要测试的视频是yuv大小大约是433M 1366x768 大约280多帧。使用CPU机器进行评分。机器内存4核32G。 运行一段时间。提示需要52G的内存。为何需要如此之大的内存?能否解答下。或者是否有参数可以调节。
首先,不建议使用CPU来跑模型,就算是解决了内存问题,CPU的运行时间会是GPU的几十倍。现在的代码是没有任何针对CPU的优化。当然,GPU模式也有很大的优化空间
直观地来说,有几个原因导致内存开销大。我们对YUV使用sliding windows采样,而且是有overlap的方式。模型运行期间也会产生高纬度的tensor,所以内存消耗比较大。另外,这是一个全参考模型,内存消耗又会加倍。数据预加载也会同步消耗内存。
在不改代码的前提下,一个比较简单的方式是减少单个测试序列的帧数,CPU/GPU内存消耗跟帧数成线性。
@haiqwang 你好,请问GPU运行情况下对显存占用有什么优化操作推荐我们去研究?我12G的显存,跑1980x1080分辨率的视频有点吃力。