ranahanocka / MeshCNN

Convolutional Neural Network for 3D meshes in PyTorch
MIT License
1.56k stars 314 forks source link

Some error I don't quite understand #83

Closed Pavelevivan closed 4 years ago

Pavelevivan commented 4 years ago

I tried to run your model on my objects for classification task using the same settings as for shrec dataset. One change is my objects have 1000 faces and I set up ninput_edges = 1500 Traceback (most recent call last): File "train.py", line 30, in model.optimize_parameters() File "/content/MeshCNN/models/mesh_classifier.py", line 66, in optimize_parameters out = self.forward() File "/content/MeshCNN/models/mesh_classifier.py", line 57, in forward out = self.net(self.edge_features, self.mesh) File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 550, in call result = self.forward(*input, kwargs) File "/usr/local/lib/python3.6/dist-packages/torch/nn/parallel/data_parallel.py", line 153, in forward return self.module(*inputs[0], *kwargs[0]) File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 550, in call result = self.forward(input, kwargs) File "/content/MeshCNN/models/networks.py", line 150, in forward x = getattr(self, 'pool{}'.format(i))(x, mesh) File "/content/MeshCNN/models/layers/mesh_pool.py", line 21, in call return self.forward(fe, meshes) File "/content/MeshCNN/models/layers/mesh_pool.py", line 34, in forward self.pool_main(mesh_index) File "/content/MeshCNN/models/layers/mesh_pool.py", line 53, in __pool_main self.pool_edge(mesh, edge_id, mask, edge_groups) File "/content/MeshCNN/models/layers/mesh_pool.py", line 61, in pool_edge elif self.clean_side(mesh, edge_id, mask, edge_groups, 0)\ File "/content/MeshCNN/models/layers/mesh_pool.py", line 79, in clean_side self.remove_triplete(mesh, mask, edge_groups, invalid_edges) File "/content/MeshCNN/models/layers/mesh_pool.py", line 181, in __remove_triplete assert(len(vertex) == 1) AssertionError

ranahanocka commented 4 years ago

Hi @Pavelevivan ,

I think your mesh is non-manifold. Did you check in meshlab? image

Pavelevivan commented 4 years ago

Yes, you are right. Thank you!