search

qubvel / segmentation_models.pytorch

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

Pretrained model tries to use segmentation_models_pytorch.unet which is not present #821

Closed fakhar-iqbal closed 8 months ago

fakhar-iqbal commented 8 months ago

I am trying to use the pretrained model by downloading it and it tries to use segmentation_models_pytorch.unet but it is not present in the repo.

fakhar-iqbal commented 8 months ago

I found this unet folder in decoder subfolder. so i copied it in the segmentation_models_pytorch folder. and that works for me, Thank you