zea.models.echonet

Echonet-Dynamic segmentation model for cardiac ultrasound segmentation. Link below does not work it seems, this is slightly different but does have some info: https://github.com/bryanhe/dynamic

Classes

EchoNetDynamic(*args, **kwargs)

EchoNet-Dynamic segmentation model for cardiac ultrasound segmentation.

class zea.models.echonet.EchoNetDynamic(*args, **kwargs)

Bases: BaseModel

EchoNet-Dynamic segmentation model for cardiac ultrasound segmentation.

Original paper and code: https://echonet.github.io/dynamic/

This class extracts useful parts of the original code and wraps it in a easy to use class.

Preprocessing should normalize the input images with mean and standard deviation.

build(input_shape)

Builds the network.

call(inputs)

Segment the input image.

custom_load_weights(preset, **kwargs)

Load the weights for the segmentation model.

maybe_convert_to_jax(input_shape)

Converts the network to Jax if backend is Jax.