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

[pjberden]

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'