NVlabs / CF-3DGS

Other
394 stars 43 forks source link

fix bug in eval_pose mode #6

Open Master-cai opened 4 months ago

Master-cai commented 4 months ago

Hi, I find a bug when performing the eval_pose mode by command

python run_cf3dgs.py --source data/Tanks/Francis \
                     --mode eval_pose \
                     --data_type tanks \
                     --model_path ./output/progressive/Tanks_Francis/chkpnt/ep00_init.pth

The error is due to directly evaluating the pose in Pytorch.Tensor, which should be Numpy.Array, in:

File "/code/CF-3DGS/utils/vis_utils.py", line 196, in plot_pose
    traj_est_aligned.align(traj_ref, correct_scale=True,

I have fixed and tested the code.

nurbanu170399 commented 4 months ago

I did the changes. Still getting the same error. Any ideas @Master-cai ?