TensoRF icon indicating copy to clipboard operation
TensoRF copied to clipboard

Test PSNR is higher than train PSNR.

Open bo233 opened this issue 2 years ago • 0 comments
trafficstars

Hi, there. Thank you for your awesome work! But when I study on your work, I meet something strange that the test PSNR is higher than train PSNR, as the following image shows. image

Specifically, the result at 30k step is:

  • test/psnr: 37.73
  • test/psnr_all: 36.51
  • train/psnr: 35.51

I check the PSNR calculation code, but everything is correct. I don't know where the problem is. I wonder whether you met this situation before.

Additionally, I train on the lego scene with the following config:


n_iters = 30000
batch_size = 4096

N_voxel_init = 2097156 # 128**3
N_voxel_final = 27000000 # 300**3
upsamp_list = [2000,3000,4000,5500,7000]
update_AlphaMask_list = [2000,4000]

N_vis = 5
vis_every = 10000

render_test = 1

n_lamb_sigma = [16,16,16]
n_lamb_sh = [48,48,48]
model_name = TensorVMSplit


shadingMode = MLP_Fea
fea2denseAct = softplus

view_pe = 2
fea_pe = 2

L1_weight_inital = 8e-5
L1_weight_rest = 4e-5
rm_weight_mask_thre = 1e-4

## please uncomment following configuration if hope to training on cp model
#model_name = TensorCP
#n_lamb_sigma = [96]
#n_lamb_sh = [288]
#N_voxel_final = 125000000 # 500**3
#L1_weight_inital = 1e-5
#L1_weight_rest = 1e-5

bo233 avatar Jul 20 '23 05:07 bo233