Resources/Guidance to incorporate tensorflow model to slicer scripted module


I developed a semantic segmentation model using U-Net (Deep learning architecture) in Tensorflow. Model is fully trained. I want to develop a scripted module in which I want to import that particular model to carry out predictions for unseen future data.

It will be very helpful if someone can refer to any reference material which helps to build the above case from scratch or any comments./ideas/guidance regarding above case are welcome

Thank you


@Fernando has done some cool tools with pytorch. Tensorflow can also be added to Slicer via pip_install so you can probably do something similar. I don’t have any example code I can share, but I’ve used tensorflow with Slicer volumes just by passing numpy arrays around.

What structures are you segmenting with the U-Net?