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

DiffusionNet for Regression? #13

Open
latmarat opened this issue Jan 6, 2022 · 2 comments
Open

DiffusionNet for Regression? #13

latmarat opened this issue Jan 6, 2022 · 2 comments

Comments

@latmarat
Copy link

latmarat commented Jan 6, 2022

Hi, very interesting work!
I am wondering if diffusion-net can be adapted for regression tasks, i.e. predicting scalars in triangles of a surface mesh?
Thanks!

@nmwsharp
Copy link
Owner

nmwsharp commented Jan 6, 2022

Yes, for sure!

You can see an example of this here, predicting values on faces. https://github.com/nmwsharp/diffusion-net/blob/master/experiments/human_segmentation_original/human_segmentation_original.py#L69-L75.

You can pass outputs_at='faces' to construct a network which outputs a value per-face. You will also need to pass in the face listing (as an Fx3 array of integers) when you invoke the newtork.

To do regression (as opposed to e.g. classification/segmentation), you can just not use any last activation from the network to output values on R^n, by omitting the last_activation argument when you build the network (by default this sets last_activation=None).

@mikkelmedm
Copy link

mikkelmedm commented Aug 8, 2022

Hi, thanks for this awesome work!
So if I wanted to predict a plane with a specific location and a certain angle intersecting the input mesh, would the network be easily configurable for that task? I.e. a simlpe regression task of rotation and translation. Or is it bound to output values 'per-face' or 'per-vertices'?
Thanks again :)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants