search

AutodeskAILab / UV-Net

Code for UV-Net: Learning from Boundary Representations, CVPR 2021.
MIT License
94 stars 16 forks source link

How to predict？ #15

Closed c464851257 closed 2 years ago

c464851257 commented 2 years ago

Thanks for sharing the code of UV-Net. I want to predict on segmentation model, but the code seems like not have predict function.

pradeep-pyro commented 2 years ago

Hi there, to get predictions from the segmentation model, you can call its forward method: https://github.com/AutodeskAILab/UV-Net/blob/91cc97042ad56b5462fb36f21959c1963147d54d/uvnet/models.py#L295-L297)

The input is a batch of DGL graphs.

c464851257 commented 2 years ago

Thank you so much.

pradeep-pyro commented 2 years ago

Great, closing this issue.