seisnn.model.generator.unet¶

seisnn.model.generator.unet(img_rows=None, img_cols=None, color_type=3, num_class=3)[source]¶

Standard U-Net [Ronneberger et.al, 2015]

Total params: 420,249

Parameters

img_rows (int) – Height of the data.
img_cols (int) – Width of the data.
color_type (int) – Channel number of the data.
num_class (int) – Output class number.

Returns

UNet model.