def effnetB4_fpncat128(num_classes=5, dropout=0.0, pretrained=True):
encoder = E.EfficientNetB4Encoder(abn_params={"activation": "swish"},
pretrained=pretrained)
return FPNCatSegmentationModel(encoder,
num_classes=num_classes,
fpn_channels=128,
dropout=dropout)
def efficientb4_fpncatv2_256(num_classes=5,
dropout=0.0,
pretrained=True,
classifiers=True):
encoder = E.EfficientNetB4Encoder()
return FPNCatSegmentationModelV2(
encoder,
num_classes=num_classes,
disaster_type_classes=len(DISASTER_TYPES) if classifiers else None,
damage_type_classes=len(DAMAGE_TYPES) if classifiers else None,
fpn_channels=256,
dropout=dropout,
abn_block=partial(ABN, activation=ACT_RELU),
)
def effnetB4_fpn(num_classes=1, fpn_features=128):
encoder = E.EfficientNetB4Encoder()
return FPNSegmentationModel(encoder, num_classes, fpn_features)