Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Encountering the issue when run this code "CUDA_VISIBLE_DEVICES=0 python run_mv.py --config configs/blender.txt --scene lego --eval_only --grad_normal" #6

Open
WuyiWong opened this issue Nov 24, 2023 · 8 comments

Comments

@WuyiWong
Copy link

hi, when I follow you tip to evaluate the ray-surface distance network through this code "CUDA_VISIBLE_DEVICES=0 python run_mv.py --config configs/blender.txt --scene lego --eval_only --grad_normal". I encounter this error:
//
Traceback (most recent call last):
File "run_mv.py", line 405, in
evaluate(args)
File "run_mv.py", line 331, in evaluate
depth = convert_d(vis_results['dist_abs'].squeeze().cpu().numpy(), dataloader.scene_info, out='dep')
KeyError: 'dist_abs'
/
/

why could this thing happen?
1700839361127

@zhuomanliu
Copy link
Member

vis_results['dist_abs'] is stored when using --denoise in the previous code. The code has been updated to fix this issue when using only --grad_normal.

@WuyiWong
Copy link
Author

WuyiWong commented Nov 25, 2023

thanks for your answer. But it seems that there is another issue I've encountered when I run this code "CUDA_VISIBLE_DEVICES=0 python run_mv.py --config configs/blender.txt --scene lego --eval_only" to evaluate the distance network.

[train] ADE=1.1995 CD=955.4302 CD_median=827.3363 PSNR=nan SSIM=nan LPIPS=nan
[test] ADE=1.1883 CD=949.7096 CD_median=835.1572 PSNR=nan SSIM=nan LPIPS=nan

why are the the values of CD, CD_median, PSNR, SSIM and LPIPS so strange, I wonder if it has anything to do with "scene lego"?

@zhuomanliu
Copy link
Member

zhuomanliu commented Nov 25, 2023

thanks for your answer. But it seems that there is another issue I've encountered when I run this code "CUDA_VISIBLE_DEVICES=0 python run_mv.py --config configs/blender.txt --scene lego --eval_only" to evaluate the distance network.

[train] ADE=1.1995 CD=955.4302 CD_median=827.3363 PSNR=nan SSIM=nan LPIPS=nan [test] ADE=1.1883 CD=949.7096 CD_median=835.1572 PSNR=nan SSIM=nan LPIPS=nan

why are the the values of CD, CD_median, PSNR, SSIM and LPIPS so strange, I wonder if it has anything to do with "scene lego"?

Have you loaded the pre-trained model during evaluation? The print log should be like this: 'Loading ckpt from: logs/multiview/...'
image

@WuyiWong
Copy link
Author

Hi,

Thanks for your patience.
It seems like the pretrained model loaded failed in my code, as shown below.
image

But I followed you evaluation steps and downloaded the checkpoints from the link you've shared. Is my step wrong?

image image

@zhuomanliu
Copy link
Member

Hi,

Thanks for your patience. It seems like the pretrained model loaded failed in my code, as shown below. image

But I followed you evaluation steps and downloaded the checkpoints from the link you've shared. Is my step wrong?

image image

How about the folder name you downloaded? It should be 'blender-lego_mv20d13w1024rgb0w1.0_lr1e-05bs8192iters80k' for Lego (geometry only). To be more specific, you can run this: "CUDA_VISIBLE_DEVICES=0 python run_mv.py --config configs/blender.txt --scene lego --eval_only --grad_normal --expname mv20d13w1024rgb0w1.0_lr1e-05bs8192iters80k" to make sure the loaded folder name is correct.

@WuyiWong
Copy link
Author

Hi, I've found the issue of my implementation. During adjustment, I've finally loaded the pre-trained model during evaluation. Thank your so much. The print log be like this: 'Loading ckpt from: logs/multiview/...'
image

But, there are still some issue in it. the value of PSNR and SSIM is really weird, is it normal? as shown below.
image

@zhuomanliu
Copy link
Member

Hi, I've found the issue of my implementation. During adjustment, I've finally loaded the pre-trained model during evaluation. Thank your so much. The print log be like this: 'Loading ckpt from: logs/multiview/...' image

But, there are still some issue in it. the value of PSNR and SSIM is really weird, is it normal? as shown below. image

It is because this is the geometry-only mode (experiment Group 1 in the paper).
For geometry+radiance mode (experiment Group 2 in the paper), adding --rgb_layer 2 is required, and then the code will load the checkpoint from the folder "scannet-scene0004_00_mv20d13w1024rgb2w1.0_lr1e-05bs8192iters80k".

@WuyiWong
Copy link
Author

I see, thank you again for your patient explanation.

Best regards.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants