lazyslide_models.segmentation.SMPBase

class SMPBase(arch='unetplusplus', encoder_name='timm-efficientnet-b0', encoder_weights='imagenet', in_channels=3, classes=3, **kwargs)

Bases: SegmentationModel

This is a base class for any models from segmentation models pytorch