search

MisEty / RTG-SLAM

RTG-SLAM: Real-time 3D Reconstruction at Scale Using Gaussian Splatting (ACM SIGGRAPH 2024)
https://gapszju.github.io/RTG-SLAM/
GNU General Public License v3.0
174 stars 12 forks source link

读取数据的过程中爆内存 #22

Closed WangPei-hitman closed 5 days ago

WangPei-hitman commented 5 days ago

image image image 大佬您好，我在运行你们hotel数据集过程中每次读取一定量的图片内存都会超过我本机32GB的容量。是不是配置上哪里有什么问题?论文中的7.3G内存消耗的数据是如何评估出来的呢？

ShreyaIAB commented 5 days ago

@WangPei-hitman 您能帮我走到这一步吗？

MisEty commented 5 days ago

7.3G是指训练过程中消耗的显存。为了提高速度我们采用了先把一次性所有图片读入内存，在训练过程中不再反复读取本地图片的方式，此处的内存消耗与算法无关。因为数据集规模较大所以消耗的内存可能较多，您可以降低分辨率或者测试规模较小的数据集replica，也可以稍微修改代码使其在训练过程中读取图片。