search

HuXiaoling / TopoLoss

Code for the NeurIPS 2019 paper: Topology-Preserving Deep Image Segmentation
MIT License
146 stars 22 forks source link

How can I use your loss function in TensorFlow #3

Closed ntelo007 closed 3 years ago

ntelo007 commented 4 years ago

Hi,

how can I implement your loss function in a TensorFlow model?

Thank you for your time.

HuXiaoling commented 4 years ago

Originally, we implement the loss function with PyTorch framework. I would recommend you try PyTorch if it's not so inconvenient:)

Of course, you can refer to https://towardsdatascience.com/custom-loss-function-in-tensorflow-2-0-d8fa35405e4e to try to incorporate the loss into tensorflow. I'm also very glad to provide any help to see how it works.