NVlabs / CF-3DGS

Other
394 stars 43 forks source link

issue with get_cmap matplotlib.cm! #10

Open brahimfarhat opened 4 months ago

brahimfarhat commented 4 months ago

Hello,

command: python run_cf3dgs.py --source data/Tanks/Francis --mode train --data_type tanks

Bug:

Downloading: "https://github.com/intel-isl/MiDaS/zipball/master" to /home/ubuntu/.cache/torch/hub/master.zip Rotation type : 6d Reading camera 150/150 Loading Training Cameras Loading Test Cameras Number of points at initialisation : 19361 Train images: 131 ['000401', '000403', '000405', '000407', '000411', '000413', '000415', '000417', '000419', '000421', '000423', '000427', '000429', '000431', '000433', '000435', '000437', '000439', '000443', '000445', '000447', '000449', '000451', '000453', '000455', '000459', '000461', '000463', '000465', '000467', '000469', '000471', '000475', '000477', '000479', '000481', '000483', '000485', '000487', '000491', '000493', '000495', '000497', '000499', '000501', '000503', '000507', '000509', '000511', '000513', '000515', '000517', '000519', '000523', '000525', '000527', '000529', '000531', '000533', '000535', '000539', '000541', '000543', '000545', '000547', '000549', '000551', '000555', '000557', '000559', '000561', '000563', '000565', '000567', '000571', '000573', '000575', '000577', '000579', '000581', '000583', '000587', '000589', '000591', '000593', '000595', '000597', '000599', '000603', '000605', '000607', '000609', '000611', '000613', '000615', '000619', '000621', '000623', '000625', '000627', '000629', '000631', '000635', '000637', '000639', '000641', '000643', '000645', '000647', '000651', '000653', '000655', '000657', '000659', '000661', '000663', '000667', '000669', '000671', '000673', '000675', '000677', '000679', '000683', '000685', '000687', '000689', '000691', '000693', '000695', '000699'] Using cache found in /home/ubuntu/.cache/torch/hub/intel-isl_MiDaS_master Using cache found in /home/ubuntu/.cache/torch/hub/intel-isl_MiDaS_master /home/ubuntu/CF-3DGS/trainer/trainer.py:491: DeprecationWarning: Since kornia 0.7.0 the depth_to_3d is deprecated in favor of depth_to_3d_v2. This function will be replaced with the depth_to_3d_v2 behaviour, where the that does not require the creation of a meshgrid. The return shape can be not backward compatible between these implementations. pts = depth_to_3d(depth_tensor[None, None], Number of points at initialisation : 272100 optimizing frame 000 Training progress: 100%|█████████████████████████████████████████████████████████████████████████████████████████████| 1000/1000 [00:22<00:00, 44.47it/s, PSNR=40.31, Number points=272100] /home/ubuntu/CF-3DGS/trainer/trainer.py:491: DeprecationWarning: Since kornia 0.7.0 the depth_to_3d is deprecated in favor of depth_to_3d_v2. This function will be replaced with the depth_to_3d_v2 behaviour, where the that does not require the creation of a meshgrid. The return shape can be not backward compatible between these implementations. pts = depth_to_3d(depth_tensor[None, None], Number of points at initialisation : 272100 Training progress: 50%|███████████████████████████████████████████████ | 500/1000 [00:11<00:11, 45.18it/s, PSNR=37.71, Number points=272100] optimizing frame 001 Training progress: 100%|███████████████████████████████████████████████████████████████████████████████████████████████| 300/300 [00:05<00:00, 54.65it/s, PSNR=30.26, Number points=272100] Training progress: 100%|███████████████████████████████████████████████████████████████████████████████████████████████| 100/100 [00:02<00:00, 41.48it/s, PSNR=37.90, Number points=272100] Frames 001/130, PSNR : 37.940 Traceback (most recent call last): File "/home/ubuntu/CF-3DGS/run_cf3dgs.py", line 47, in trainer.train_from_progressive() File "/home/ubuntu/CF-3DGS/trainer/cf3dgs_trainer.py", line 474, in train_from_progressive self.visualize(render_dict, File "/home/ubuntu/CF-3DGS/trainer/cf3dgs_trainer.py", line 796, in visualize colorize(render_pkg["depth"].detach().cpu().numpy(), File "/home/ubuntu/CF-3DGS/utils/image_utils.py", line 128, in colorize cmapper = mpl.cm.get_cmap(cmap) AttributeError: module 'matplotlib.cm' has no attribute 'get_cmap'

brahimfarhat commented 4 months ago

it is working with matplotlib v 3.3.2

pip install matplotlib==3..3.2

Rashfu commented 4 months ago

Can you show me the version of your evo package? Installing the missing pip package automatically crashes the code.

brahimfarhat commented 4 months ago

Can you show me the version of your evo package? Installing the missing pip package automatically crashes the code.

yes, evo = 1.29.0