-
Notifications
You must be signed in to change notification settings - Fork 654
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
model name #345
Comments
Good questions!
Yes, MONAI models are imported here:
If you want to change the code of a model you can create a new model class in unet.py and change the name in the configuration file to use that class. If you want to specifically modify the DynUNet model, you could create a new class that inherits the DynUnet model from MONAI and modify select parts to your choosing. You can also use the UNet models that I wrote for this project and play around with the code for those, but I don't think they perform as well as a some of the MONAI models, so that is why the DynUNet is the default for the example. |
Thank you, author |
Hello author, I have another question. What is the configuration of your computer? |
Yes, segmentation.py is where the data is loaded using MONAI functions.
See line 53 where it resizes the image to the desired shape. Depending on the configuration parameters, it can also crop/pad the image to the desired shape, or randomly crop the image to the desired shape. |
That is odd that the memory would explode for prediction rather than training. Prediction should be relatively easy compared to doing all the back propagation during training. I can train the model Brats model using a rather old GPU with 11GB of memory. |
|
load_dataset_class loads the dataset according to the name specified in the configuration file. If the configuration file specifies the "SegmentationDatasetPersistent" then it will load the class defined in the segmentation.py file. You can also write your own dataset class and specify that in your configuration file instead. |
Hello author, sorry to disturb you
I would like to ask how the model is imported. If I want to modify the model, where should I go to modify it? I couldn't find a model called DynUNet. Did you import it from the MONAI package?
The text was updated successfully, but these errors were encountered: