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]

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

ALIASES

sigmoid_threshold