Skip to content

Spatial Models API

deepuq.models.spatial

CNNRegressor2D

Bases: Module

Compact image-to-image CNN baseline for 2D scientific fields.

ResNetRegressor2D

Bases: Module

Residual 2D field-to-field regressor with optional spatial dropout.

UNet2D

Bases: Module

2D U-Net backbone for field-to-field scientific regression.

UNet3D

Bases: Module

3D U-Net backbone for volumetric scientific regression.