lucaskre / dilated_tooth_seg_net Goto Github PK

Hello, what is the function of the point cloud operation library in the code, and why do you need to write an additional pointnet structure?

Hello
I met this question in running this code
he value of the num_workers argumenttonum_workers=11in theDataLoader` to improve performance.
Sanity Checking DataLoader 0: 0%| | 0/2 [00:00<?, ?it/s]Traceback (most recent call last):
File "/mnt/dilated_tooth_seg_net-main/train_network.py", line 102, in
trainer.fit(model=model, train_dataloaders=train_dataloader, val_dataloaders=val_dataloader, ckpt_path=args.ckpt)
File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/trainer/trainer.py", line 545, in fit
call._call_and_handle_interrupt(
File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/trainer/call.py", line 44, in _call_and_handle_interrupt
return trainer_fn(*args, **kwargs)
File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/trainer/trainer.py", line 581, in _fit_impl
self._run(model, ckpt_path=ckpt_path)
File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/trainer/trainer.py", line 990, in _run
results = self._run_stage()
File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/trainer/trainer.py", line 1034, in _run_stage
self._run_sanity_check()
File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/trainer/trainer.py", line 1063, in _run_sanity_check
val_loop.run()
File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/loops/utilities.py", line 181, in _decorator
return loop_run(self, *args, **kwargs)
File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/loops/evaluation_loop.py", line 134, in run
self._evaluation_step(batch, batch_idx, dataloader_idx, dataloader_iter)
File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/loops/evaluation_loop.py", line 391, in _evaluation_step
output = call._call_strategy_hook(trainer, hook_name, *step_args)
File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/trainer/call.py", line 309, in _call_strategy_hook
output = fn(*args, **kwargs)
File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/strategies/strategy.py", line 403, in validation_step
return self.lightning_module.validation_step(*args, **kwargs)
File "/mnt/dilated_tooth_seg_net-main/models/dilated_tooth_seg_network.py", line 109, in validation_step
pred = self.model(x, pos)
File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1518, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1527, in _call_impl
return forward_call(*args, **kwargs)
File "/mnt/dilated_tooth_seg_net-main/models/dilated_tooth_seg_network.py", line 63, in forward
x1, _ = self.dilated_edge_graph_conv_block1(x, pos, cd=cd)
File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1518, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1527, in _call_impl
return forward_call(*args, **kwargs)
File "/mnt/dilated_tooth_seg_net-main/models/layer.py", line 229, in forward
idx_fps = fps(pos.reshape(B * N, -1)[idx_l], self.k).long()
NameError: name 'fps' is not defined

How can I solve it
It happpens after processing data

Could you consider providing the code to visualize the segmentation results? Just like the figure shows in the paper and Readme markdown. Thanks.