segmentation

smp_resnet18_unet()

Yield a unet with resnet18 encoder.

Source code in quadra/utils/tests/fixtures/models/segmentation.py

@pytest.fixture
def smp_resnet18_unet():
    """Yield a unet with resnet18 encoder."""
    yield create_smp_backbone(
        arch="unet",
        encoder_name="resnet18",
        encoder_weights=None,
        encoder_depth=5,
        freeze_encoder=True,
        in_channels=3,
        num_classes=1,
        activation=None,
    )

smp_resnet18_unetplusplus()

Yield a unetplusplus with resnet18 encoder.

Source code in quadra/utils/tests/fixtures/models/segmentation.py

@pytest.fixture
def smp_resnet18_unetplusplus():
    """Yield a unetplusplus with resnet18 encoder."""
    yield create_smp_backbone(
        arch="unetplusplus",
        encoder_name="resnet18",
        encoder_weights=None,
        encoder_depth=5,
        freeze_encoder=True,
        in_channels=3,
        num_classes=1,
        activation=None,
    )