qubvel-org / segmentation_models.pytorch

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

Update timm to 1.* version and support more encoders #885

Open qubvel opened 5 months ago

qubvel commented 5 months ago

Make most of the models from timm compatible with decoders (including transformers and convnext models)

qubvel commented 5 months ago

Hi @wouterzwerink thanks for the review! I am working on it, it requires a bit more in depth analysis to make most of the models compatible and configurable.

adamjstewart commented 2 months ago

According to #918, timm 1.0.9 already works great. Or our tests are useless. One or the other.

qubvel commented 2 months ago

This PR is more about unlocking newer backbones, the latest timm itself should work with the current setup