3D Gaussian Splatting训练时一次只读入一张图片进行训练

3D Gaussian Splatting训练时,原代码会将所有相机相关参数和对应图片都读入内存中,如果训练的图片尺寸过大且数量过多,那么只是在读入数据时进程就会因占用过大内存而被系统kill。

为了能训练过大尺寸的图片且不进行下采样,就需要修改原代码,在3D Gaussian Splatting每次训练时只读入抽取到的相机参数和图片进行训练。

既然是在每次训练时才随机读入一个相机,那么原来的Scene的初始化函数就要修改,只进行与相机参数无关的数据初始化。

其中需要注意的是self.cameras_extent这个值,需要根据images.txt文件里的相机外参计算出来填入。

既然在初始化时没有读入相机,那么就需要另写一个函数来实现根据随机抽取到的相机id来读入对应的相机参数了。

具体的读入相机参数和对应图片函数修改如下:

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值