解决方案很简单。SMP模型有额外的参数aux_params，如文档中所述。分类头参数可以通过它传递。

aux_params=dict(
    pooling='avg',             # one of 'avg', 'max'
    activation='sigmoid',      # activation function, default is None
    classes=1,                 # define number of output labels
  )

 model = smp.MAnet(
     encoder_name="efficientnet-b0", 
     encoder_weights="imagenet",    
     in_channels=3,                 
     classes=1,                    
     aux_params = aux_params
 )

字符串
完成此设置后，model将返回一个元组，其中包含用于分类和分割的输出。