NeRF不同阶段时间消耗测试结果

测试设备:Windows 11 专业版
显卡型号:NVIDIA 3050
显存:4G
CPU型号: Intel 12th Gen i5-12450H
内存:16G
测试代码https://github.com/xunull/read-nerf-pytorch
测试数据集描述:自建数据集,原始图像大小:1280 px X 720 px,测试过程中使用下采样数据集,下采样2x之后的图像大小为:640px X 360px。数据集中一共包含:113张图片。使用colmap进行位姿恢复。数据格式为:LLFF。

参数设置

#实验名称和数据集位置
expname = xxx_test
basedir = ./logs
datadir = ./data/llff/test
dataset_type = llff

# 下采样倍数
factor = 2
llffhold = 2

#批量大小1024根光线
N_rand = 1024
N_samples = 64
N_importance = 64

use_viewdirs = True
raw_noise_std = 1e0

spherify = True
render_only = True

1、训练过程:

(1)数据准备过程

(2)主循环(20000次迭代,每一次迭代是一个批量的数据)

3、结论分析:

在数据准备的过程中,将数据集重构最花费时间,将图像数据构建为[(N-3)*H*W, ro+rd+rgb, 3],所需要的时间最多。

然而在训练过程中,最花费时间的是主循环的内容,主循环包括了:获取数据、前向推理和反向传播三个环节。在评估推理时间时,我们将保存模型、测试视频和测试图片的时间出去。当每完成一个批次的训练,将会花费1.3秒钟左右,其中模型前向推理的时间最多,占比81%。如果主循环轮次为20000轮,则可估计出在该电脑配置上,将需要7.2小时完成训练。

在测试过程中,每一个的相机位姿需要10.67秒推理一张图片,所以完成120个相机位姿的计算总共需要花费21.2分钟。经排查,测试过程中最花费时间的仍然是前向推理过程,训练过程中一个批次只有1024条光线参与推理,形状为:[1024, 11]。但是在测试的时候,一张图像存在640*360个像素,所以,输入的数据的形状是:[230400, 11]。输入的数据量更大,自然需要更多的时间。

关于NeRF神经网络的速度优化可以从以下几方面进行优化:
(1)利用instant-ngp、3D高斯等技术加速训练过程。
(2)利用神经网络压缩技术对网络结构进行优化
(3)可以考虑利用maching cubes技术将隐式模型,转换为mesh、点云、obj等格式。可直接获得整体模型,但可能会导致模型精度变差。
(4)不使用colmap生成位姿,由slam直接获取位姿。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,732评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 87,496评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,264评论 0 338
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,807评论 1 277
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,806评论 5 368
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,675评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,029评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,683评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 41,704评论 1 299
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,666评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,773评论 1 332
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,413评论 4 321
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,016评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,978评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,204评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,083评论 2 350
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,503评论 2 343

推荐阅读更多精彩内容