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)

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