how to use UnetPlusPlus

qubvel-org / segmentation_models.pytorch

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

https://smp.readthedocs.io/

MIT License

9.58k stars 1.67k forks source link

Closed picEmily closed 3 years ago

picEmily commented 3 years ago

model = smp.UnetPlusPlus()

AttributeError: module 'segmentation_models_pytorch' has no attribute 'UnetPlusPlus'

qubvel commented 3 years ago

Hi, It is not yet in pip release, install from source

picEmily commented 3 years ago

Thank you