NeRF(Neural Radiace Fields)是一种最新的三维新视角合成技术。其核心利用物理体渲染(Volume rederig)方程结合各类隐式表征的方式,使模型根据三维物体或场景的二维多视角彩色图像即可完成高质量的三维重建。 现有各类 NeRF 方法主要针对低分辨率场景(<=1K)设计,在更高分辨率的场景上就会出现渲染模糊,细节丢失,推理速度缓慢等问题。4K-NeRF 是针对以上问题设计的,在超高分辨率上可以取得相比现有方法明显的效果提升。 4K-NeRF 将 NeRF 结构改进为编码-解码器结构,采用显式表示的 NeRF 模型作为编码器,主要学习低分辨下的几何信息;而为了高效的恢复渲染高分辨细节,解码器则采用预训练的图像恢复卷积网络。同时采用基于邻域的光线采样方式结合基于图像感知的损失函数。 以下分别是在 erfllffdata 和 erf_sythetic 上 4K-NeRF 与 baselie 的渲染效果对比。 当前支持两种数据类型,分别是erf-sythesis 和 erfllffdata。 在 ModelScope 框架上,调用训练好的 4K-NeRF 模型即可通过简单的 pipelie 来推理(仅支持 GPU 运行)。 以 erfllffdata 的 fer 场景为例,在构建 pipelie 时需要指定 ecckptpath 和 decckptpath 分别为训练好的 ecoder 和 decoder 的模型路径。除此之外,pipelie 构建完成后需要输入 data_dic 来提供数据集相关的参数。具体示例可以参考以下代码,该代码可直接运行: 训练代码和训练方式请参考 [github] Zhogshu Wag, Ligzhi Li, Zhe She, Li She, Liefeg Bo Alibaba GroupCloe with HTTP
git cloe https://www.modelscope.c/DAMOXR/cv_erf-3d-recostructio-4k-erf_damo.git
背景介绍
模型概述
效果展示
使用范围与方式
使用范围
目标场景
运行环境
使用方式
模型推理
import os
from modelscope.pipelies import pipelie
from modelscope.utils.costat import Tasks
from modelscope.msdatasets import MsDataset
#
model_id = 'DAMOXR/cv_erf-3d-recostructio-4k-erf_damo'
data_dir = MsDataset.load(
'DAMOXR/erf_llff_data',
subset_ame='default',
split='test',
# dowload_mode=DowloadMode.FORCE_REDOWNLOAD
).cofig_kwargs['split_cofig']['test']
erf_llff = os.path.joi(data_dir, 'erf_llff_data')
scee = 'fer'
data_dir = os.path.joi(erf_llff, scee)
reder_dir = 'exp'
data_dic = dict(
datadir=data_dir,
dataset_type='llff',
load_sr=1,
factor=4,
dc=True,
white_bkgd=False)
reder_dir = 'exp'
### ow we use llff dataset, whe use erf-sythesis dataset, data_type should specify as 'bleder'
erf_reco_4k = pipelie(
Tasks.erf_reco_4k,
model=model_id,
data_type='llff',
)
erf_reco_4k(
dict(data_cfg=data_dic, reder_dir=reder_dir))
### reder results will be saved i reder_dir
模型训练
相关论文
4K-NeRF: High Fidelity Neural Radiace Fields at Ultra High Resolutios
点击空白处退出提示







评论