ananyajana/tooth_mesh_seg

Inference problem

Opened this issue · 0 comments

Thank you very much for your work. It is very interesting.

I have tried to perform inference on the test with the model you propose in this repository. For this, I have made use of the repository https://github.com/ananyajana/single_dental_model_learning where you use the pointMLP model. Therefore I have tried to replicate that inference that I have managed to run for this new model but I get the following error.

Se produjo una excepción: IndexError Dimension out of range (expected to be in range of [-2, 1], but got 2) File "curvenet_util.py", line 466, in forward start_index = start_index.squeeze().unsqueeze(2) File "curvenet_util.py", line 366, in forward curves = self.curvegrouping(x, xyz, idx[:,:,1:]) # avoid self-loop File "pointMLP.py", line 563, in forward l1_xyz, l1_points = self.cic11(xyz, l0_points) File "predict_on_test_setPCURVE.py", line 288, in <module> tensor_prob_output = model(input_DATA) IndexError: Dimension out of range (expected to be in range of [-2, 1], but got 2)

What could be the problem?