pyg-team / pytorch_geometric

Graph Neural Network Library for PyTorch
https://pyg.org
MIT License
21.29k stars 3.65k forks source link

DGCNN and PointNet++ add expected segmentation results to examples #849

Open AruniRC opened 4 years ago

AruniRC commented 4 years ago

❓ Questions & Help

Can you please mention the expected mean IoU over all ShapeNet segmentation categories using the example code you provide for DGCNN and PointNet segmentation?

The current code example does not provide any indication of how close the results from this code is with respect to the reported numbers in the two papers. It would be really very helpful to have these as a sanity-check when using your models and codebase as a starting point for any project.

thank you.

rusty1s commented 4 years ago

This is a great idea! In addition, we might provide pre-trained models for those models. I will re-run the experiments and add the expected results when I have time.

AruniRC commented 4 years ago

Hi @rusty1s, a heads-up -- I am trying to reproduce the results for PointNet++ for ShapeNet segmentation, as a first step. A quick recap:

Will let you know on this thread if I can resolve the remaining "deltas" and get comparable numbers to the published results.

rusty1s commented 4 years ago

Thank you for your investigations! I think we should add normals to the ShapeNet dataset which should bring huge gains IMO.

Tofull commented 4 years ago

normals or any additional n-dimensions (like intensity for lidar point clouds)

tchaton commented 4 years ago

Dear @AruniRC @Tofull @kolia @fadel,

I started to build a proper benchmark: https://github.com/nicolas-chaulet/deeppointcloud-benchmarks

Please, have a look :)

It is still under heavy dev.

Best to All,