search

DQiaole / MemFlow

[CVPR 2024] MemFlow: Optical Flow Estimation and Prediction with Memory
Apache License 2.0
114 stars 6 forks source link

请问支持最大的输入图像是多少呢 #8

Closed WorldHellooo closed 2 months ago

WorldHellooo commented 2 months ago

观察到有些尺寸会让gpu内存溢出

DQiaole commented 2 months ago

我用80G的A100跑1080p的视频是没问题的。至于具体的尺寸上限，不同GPU都不太一样。如果出现GPU OOM的情况，推荐降低输入分辨率，最后再把输出的flow插值到原分辨率。