boundary in pytorch implementation

LiyaoTang / contrastBoundary

Contrastive Boundary Learning for Point Cloud Segmentation (CVPR2022)

MIT License

139 stars 11 forks source link

Closed cb6e310 closed 8 months ago

cb6e310 commented 1 year ago

Hi, Is boundary computation implemented in PyTorch? I was so far not able to find any.

I appreciate any help you can provide.

LiyaoTang commented 8 months ago

Hi,

Thanks for your interest.

Yes, it is. You can find it at get_boundary_mask. For the model inference, it is directly implemented in-place here.

Best, Liyao