huiyu-gao / VisFusion

[CVPR 2023] Code for "VisFusion: Visibility-aware Online 3D Scene Reconstruction from Videos"
https://huiyu-gao.github.io/visfusion
Apache License 2.0
181 stars 9 forks source link

IndexError: max(): Expected reduction dim 0 to have non-zero size. #5

Open yyj2518 opened 1 year ago

yyj2518 commented 1 year ago

Really excellent work!

When I run the code, I can not get the reconstruction result . I found a error :

Traceback (most recent call last): File "main.py", line 337, in test() File "main.py", line 270, in test loss, scalar_outputs, outputs = test_sample(sample, save_scene) File "/home/sdu/yyj/VisFusion-main/utils.py", line 27, in wrapper ret = func(*f_args, f_kwargs) File "main.py", line 327, in test_sample outputs, loss_dict = model(sample, save_scene) File "/home/sdu/anaconda3/envs/neucon/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1130, in _call_impl return forward_call(*input, *kwargs) File "/home/sdu/anaconda3/envs/neucon/lib/python3.7/site-packages/torch/nn/parallel/data_parallel.py", line 166, in forward return self.module(inputs[0], kwargs[0]) File "/home/sdu/anaconda3/envs/neucon/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1130, in _call_impl return forward_call(*input, *kwargs) File "/home/sdu/yyj/VisFusion-main/models/visfusion_net.py", line 89, in forward outputs=outputs, save_mesh=save_mesh) File "/home/sdu/anaconda3/envs/neucon/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1130, in _call_impl return forward_call(input, **kwargs) File "/home/sdu/yyj/VisFusion-main/models/global_fusion.py", line 406, in forward outputs = self.save_mesh(layer, outputs, self.scene_name[layer]) File "/home/sdu/yyj/VisFusion-main/models/global_fusion.py", line 206, in save_mesh max_c = torch.max(fuse_coords, dim=0)[0][:3] IndexError: max(): Expected reduction dim 0 to have non-zero size.

Could you help me to solve it. thx!