search

YudongGuo / AD-NeRF

This repository contains a PyTorch implementation of "AD-NeRF: Audio Driven Neural Radiance Fields for Talking Head Synthesis".
MIT License
1.03k stars 175 forks source link

RTX3060 合成一帧要 24 秒，合成 300 帧需要 2 小时，这正常吗 #123

Open Hybrogen opened 1 year ago

Hybrogen commented 1 year ago

您好，很棒的项目！合成结果确实清晰匹配效果也好。 我这个电脑如标题所示，因为运行时 CPU 只用了不到 40% 而 GPU 拉满，根据性能对比，数据预处理和训练的速度与网上描述差不多。但是没有找到关于本项目合成时间的描述。 请问 Yudong 可不可以提供一下您的运行配置以及合成消耗的时间，我很好奇我这个运行时间是否还有优化的可能。

Hybrogen commented 1 year ago

使用原本的命令合成需要 2 小时 50 分钟左右，我修改了一些参数如下：

python NeRFs/TorsoNeRF/run_nerf.py --config dataset/Obama/TorsoNeRFTest_config.txt --aud_file=dataset/Obama/aud.npy --test_size=100 --N_samples 32 --chunk 16384

可以使本来一个进度条 17 s 加快到一个进度条 12 s，但依然感觉合成速度并不是很快。所以想咨询一下各位的合成时间是大概是多少。

feifeitu1 commented 1 year ago

请问一下合成的视频最后在哪个文件啊？