| library_name: segmentation-models-pytorch | |
| license: mit | |
| pipeline_tag: image-segmentation | |
| tags: | |
| - model_hub_mixin | |
| - pytorch_model_hub_mixin | |
| - segmentation-models-pytorch | |
| - semantic-segmentation | |
| - pytorch | |
| languages: | |
| - python | |
| # UPerNet Model Card | |
| Table of Contents: | |
| - [Load trained model](#load-trained-model) | |
| - [Model init parameters](#model-init-parameters) | |
| - [Model metrics](#model-metrics) | |
| - [Dataset](#dataset) | |
| ## Load trained model | |
| ```python | |
| import segmentation_models_pytorch as smp | |
| model = smp.from_pretrained("<save-directory-or-this-repo>") | |
| ``` | |
| ## Model init parameters | |
| ```python | |
| model_init_params = { | |
| "encoder_name": "tu-resnet18", | |
| "encoder_depth": 5, | |
| "encoder_weights": "imagenet", | |
| "decoder_channels": 256, | |
| "decoder_use_norm": "batchnorm", | |
| "in_channels": 3, | |
| "classes": 1, | |
| "activation": None, | |
| "upsampling": 4, | |
| "aux_params": None | |
| } | |
| ``` | |
| ## Model metrics | |
| [More Information Needed] | |
| ## Dataset | |
| Dataset name: [More Information Needed] | |
| ## More Information | |
| - Library: https://github.com/qubvel/segmentation_models.pytorch | |
| - Docs: https://smp.readthedocs.io/en/latest/ | |
| This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) |