dosma.models.IWOAIOAIUnet2D
- class dosma.models.IWOAIOAIUnet2D(input_shape, weights_path, force_weights=False)[source]
Model trained by Team 6 in the 2019 IWOAI Segmentation Challenge.
References
Desai, et al., “The International Workshop on Osteoarthritis Imaging Knee MRI Segmentation Challenge: A Multi-Institute Evaluation and Analysis Framework on a Standardized Dataset.” arXiv preprint arXiv:2004.14003 (2020). [link]
Methods
__init__(input_shape, weights_path[, ...])build_model(input_shape[, weights_path])Builds a segmentation model architecture and loads weights.
generate_mask(volume)Segment the MRI volumes.
Attributes
ALIASESsigmoid_threshold