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
-