search

qubvel-org / segmentation_models.pytorch

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
https://smp.readthedocs.io/
MIT License
9.73k stars 1.68k forks source link

TypeError: __init__() got an unexpected keyword argument 'output_stride' #978

Open LuShuaie opened 1 week ago

LuShuaie commented 1 week ago 
smp.DeepLabV3Plus(encoder_name='tu-maxvit_base_tf_224',
                  encoder_depth=5, encoder_weights='imagenet',
                  encoder_output_stride=16, decoder_channels=256,
                  decoder_atrous_rates=(12, 24, 36), in_channels=3,
                  classes=1, activation=None, upsampling=4, aux_params=None)

when I run the above code,I get error, TypeError: init() got an unexpected keyword argument 'output_stride'

brianhou0208 commented 1 week ago

Hi @LuShuaie , maxvit doesn't support the output_stride parameter due to lack of dilation support. Check the doc for more info.