3D Gaussian Splatting训练时,原代码会将所有相机相关参数和对应图片都读入内存中,如果训练的图片尺寸过大且数量过多,那么只是在读入数据时进程就会因占用过大内存而被系统kill。
为了能训练过大尺寸的图片且不进行下采样,就需要修改原代码,在3D Gaussian Splatting每次训练时只读入抽取到的相机参数和图片进行训练。
既然是在每次训练时才随机读入一个相机,那么原来的Scene的初始化函数就要修改,只进行与相机参数无关的数据初始化。
其中需要注意的是self.cameras_extent这个值,需要根据images.txt文件里的相机外参计算出来填入。
既然在初始化时没有读入相机,那么就需要另写一个函数来实现根据随机抽取到的相机id来读入对应的相机参数了。
具体的读入相机参数和对应图片函数修改如下: