ethnhe / FFB6D

[CVPR2021 Oral] FFB6D: A Full Flow Bidirectional Fusion Network for 6D Pose Estimation.
MIT License
290 stars 72 forks source link

Demo_lm.sh error: 'nearest_neighbors' has no attribute 'knn_bat #93

Open pjberden opened 1 year ago

pjberden commented 1 year ago

I ran "sh demo_lm.sh" which resulted in this error: AttributeError: module 'nearest_neighbors' has no attribute 'knn_batch'.

Output of terminal is:

(ffb6d) sander@LAPTOP-8QQLID97:~/FFB6D/ffb6d$ sh demo_lm.sh mkdir: cannot create directory ‘/home/sander/FFB6D/ffb6d/train_log/linemod/eval_results/’: Permission denied mkdir: cannot create directory ‘/home/sander/FFB6D/ffb6d/train_log/linemod/train_info/’: Permission denied /home/sander/FFB6D/ffb6d/common.py:139: YAMLLoadWarning: calling yaml.load() without Loader=... is deprecated, as the default Loader is unsafe. Please read https://msg.pyyaml.org/load for full details. self.lm_r_lst = yaml.load(lm_r_file) mkdir: cannot create directory ‘/home/sander/FFB6D/ffb6d/train_log/linemod/eval_results’: Permission denied mkdir: cannot create directory ‘/home/sander/FFB6D/ffb6d/train_log/linemod/train_info’: Permission denied mkdir: cannot create directory ‘/home/sander/FFB6D/ffb6d/train_log/linemod/eval_results’: Permission denied mkdir: cannot create directory ‘/home/sander/FFB6D/ffb6d/train_log/linemod/train_info’: Permission denied cls_id in lm_dataset.py 1 /home/sander/FFB6D/ffb6d/datasets/linemod/linemod_dataset.py:46: YAMLLoadWarning: calling yaml.load() without Loader=... is deprecated, as the default Loader is unsafe. Please read https://msg.pyyaml.org/load for full details. self.meta_lst = yaml.load(meta_file) test_dataset_size: 1050 /home/sander/anaconda3/envs/ffb6d/lib/python3.7/site-packages/torch/utils/data/dataloader.py:557: UserWarning: This DataLoader will create 20 worker processes in total. Our suggested max number of worker in current system is 8, which is smaller than what this DataLoader is going to create. Please be aware that excessive worker creation might get DataLoader running slow or even freeze, lower the worker number to avoid potential slowness/freeze if necessary. cpuset_checked)) loading resnet34 pretrained mdl. ==> Loading from checkpoint 'train_log/linemod/checkpoints/ape/FFB6D_ape_best.pth.tar' ==> Done val: 0it [00:00, ?it/s]kps_pth in get_kps: datasets/linemod/kps_orb9_fps/ape_8_kps.txt kps_pth in get_kps: datasets/linemod/kps_orb9_fps/ape_8_kps.txt kps_pth in get_kps: datasets/linemod/kps_orb9_fps/ape_8_kps.txt kps_pth in get_kps: datasets/linemod/kps_orb9_fps/ape_8_kps.txt kps_pth in get_kps: datasets/linemod/kps_orb9_fps/ape_8_kps.txt kps_pth in get_kps: datasets/linemod/kps_orb9_fps/ape_8_kps.txt kps_pth in get_kps: datasets/linemod/kps_orb9_fps/ape_8_kps.txt kps_pth in get_kps: datasets/linemod/kps_orb9_fps/ape_8_kps.txt kps_pth in get_kps: datasets/linemod/kps_orb9_fps/ape_8_kps.txt kps_pth in get_kps: datasets/linemod/kps_orb9_fps/ape_8_kps.txt kps_pth in get_kps: datasets/linemod/kps_orb9_fps/ape_8_kps.txt kps_pth in get_kps: datasets/linemod/kps_orb9_fps/ape_8_kps.txt kps_pth in get_kps: datasets/linemod/kps_orb9_fps/ape_8_kps.txt kps_pth in get_kps: datasets/linemod/kps_orb9_fps/ape_8_kps.txt kps_pth in get_kps: datasets/linemod/kps_orb9_fps/ape_8_kps.txt kps_pth in get_kps: datasets/linemod/kps_orb9_fps/ape_8_kps.txt kps_pth in get_kps: datasets/linemod/kps_orb9_fps/ape_8_kps.txt kps_pth in get_kps: datasets/linemod/kps_orb9_fps/ape_8_kps.txt kps_pth in get_kps: datasets/linemod/kps_orb9_fps/ape_8_kps.txt kps_pth in get_kps: datasets/linemod/kps_orb9_fps/ape_8_kps.txt Traceback (most recent call last): File "/home/sander/anaconda3/envs/ffb6d/lib/python3.7/runpy.py", line 193, in _run_module_as_main "main", mod_spec) File "/home/sander/anaconda3/envs/ffb6d/lib/python3.7/runpy.py", line 85, in _run_code exec(code, run_globals) File "/home/sander/FFB6D/ffb6d/demo.py", line 186, in main() File "/home/sander/FFB6D/ffb6d/demo.py", line 180, in main enumerate(test_loader), leave=False, desc="val" File "/home/sander/anaconda3/envs/ffb6d/lib/python3.7/site-packages/tqdm/std.py", line 1195, in iter for obj in iterable: File "/home/sander/anaconda3/envs/ffb6d/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 628, in next data = self._next_data() File "/home/sander/anaconda3/envs/ffb6d/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 1333, in _next_data return self._process_data(data) File "/home/sander/anaconda3/envs/ffb6d/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 1359, in _process_data data.reraise() File "/home/sander/anaconda3/envs/ffb6d/lib/python3.7/site-packages/torch/_utils.py", line 543, in reraise raise exception AttributeError: Caught AttributeError in DataLoader worker process 0. Original Traceback (most recent call last): File "/home/sander/anaconda3/envs/ffb6d/lib/python3.7/site-packages/torch/utils/data/_utils/worker.py", line 302, in _worker_loop data = fetcher.fetch(index) File "/home/sander/anaconda3/envs/ffb6d/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 58, in fetch data = [self.dataset[idx] for idx in possibly_batched_index] File "/home/sander/anaconda3/envs/ffb6d/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 58, in data = [self.dataset[idx] for idx in possibly_batched_index] File "/home/sander/FFB6D/ffb6d/datasets/linemod/linemod_dataset.py", line 451, in getitem return self.get_item(item_name) File "/home/sander/FFB6D/ffb6d/datasets/linemod/linemod_dataset.py", line 320, in get_item cld[None, ...], cld[None, ...], 16 File "/home/sander/FFB6D/ffb6d/models/RandLA/helper_tool.py", line 170, in knn_search neighbor_idx = nearest_neighbors.knn_batch(support_pts, query_pts, k, omp=True) AttributeError: module 'nearest_neighbors' has no attribute 'knn_batch'