Hi, I’m working on a project about pelvic floor MRI image recognition. In this project I am trying to automatic recognize feature points and measure distance on MRI images which will help with pelvic floor disease detection. Right now I have already trained a neural network with Keras and stored the model in a .h5 file. Since Slicer is the perfect open-source platform to implement this idea, I have been working hard to learn, but still a beginner to Slicer. I have finished building the GUI of a scripted module.Now I am trying to put my model into the scripted module I created. And I’m seeking advices and ideas about how to do this.
I’m now using
from pip._internal import main as _main
to load packages to Slicer and
model = load_model(’./models/’ + model_name + ‘.h5’,compile=False)
to load my module.
I’m wondering if my idea is ok and is there a better way to do it?
Does anyone have any suggestions?