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]

__init__(input_shape, weights_path, force_weights=False)[source]¶: Initialize self. See help(type(self)) for accurate signature.

Methods

`__init__`(input_shape, weights_path[, …])	Initialize self.
`build_model`(input_shape, weights_path)	Builds a segmentation model architecture and loads weights.
`generate_mask`(volume)	Segment the MRI volumes.

Attributes

`ALIASES`
`sigmoid_threshold`