search

isaaccorley / torchseg

Segmentation models with pretrained backbones. PyTorch.
MIT License
104 stars 8 forks source link

Example with other Decoders and Encoders #28

Open bach05 opened 6 months ago

bach05 commented 6 months ago

Hello, I could use the models you reported in the examples, like Unet + swin_tiny_patch4_window7_224 (or similar). But when I try to change the decoder (like DeepLabV3Plus) it is really tough to understand how to set the parameters.

The same foe encoders. If I try to use deit3_large_patch16_224, it is really difficult to understand how to make it works.

Could you please provide more example or a guide?

isaaccorley commented 6 months ago

@bach05 I will add an example to the readme. It's a bit tricky with deeplab because it requires the encoder features to be a specific size with output_stride=8. This requires setting the scale_factors properly.