dosma.models.OAIUnet2D

class dosma.models.OAIUnet2D(input_shape, weights_path, force_weights=False)[source]

Model trained in Chaudhari et al. IWOAI 2018

Original Github: https://github.com/akshaysc/msk_segmentation

__init__(input_shape, weights_path, force_weights=False)

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