Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

add pointnet++ part segmentation example #285

Merged
merged 3 commits into from May 15, 2019

Conversation

Projects
None yet
2 participants
@dragonbook
Copy link
Contributor

commented May 9, 2019

Hi, @rusty1s

This is a pointnet++ part segmentation example as I mentioned at issue141. Sorry for the delay and my poor code comment.

@rusty1s

This comment has been minimized.

Copy link
Owner

commented May 9, 2019

Really cool. I will have a closer look next week. Sorry for the delay.

@dragonbook

This comment has been minimized.

Copy link
Contributor Author

commented May 15, 2019

I tested the implementation on shapenet, and it can perform well.

Performance

Segmentation on A subset of shapenet.

Method mcIoU Airplane Bag Cap Car Chair Earphone Guitar Knife Lamp Laptop Motorbike Mug Pistol Rocket Skateboard Table
PointNet++ 81.9 82.4 79.0 87.7 77.3 90.8 71.8 91.0 85.9 83.7 95.3 71.6 94.1 81.3 58.7 76.4 82.6
PointCNN 84.6 84.11 86.47 86.04 80.83 90.62 79.70 92.32 88.44 85.31 96.11 77.20 95.28 84.21 64.23 80.00 82.99
PointNet++(this repo) 84.68 85.42 85.92 88.39 79.73 91.86 75.37 92.95 88.56 85.72 97.00 72.94 96.88 84.52 64.38 79.39 85.91

mcIOU: mean per-class pIoU

@rusty1s

This comment has been minimized.

Copy link
Owner

commented May 15, 2019

Hi, I tested it myself and it runs great. You rock. There are a few things I would do differently (like outsourcing some functionality to PyG, e.g., IoU computation), but I can do this in the master branch. Again, thank you so much for your valuable contribution.

@rusty1s rusty1s merged commit 08f98cf into rusty1s:master May 15, 2019

1 check failed

continuous-integration/travis-ci/pr The Travis CI build failed
Details
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.