Water Meter Segmentation

--- 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 --- # Unet 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": "resnet50", "encoder_depth": 5, "encoder_weights": "imagenet", "decoder_use_batchnorm": True, "decoder_channels": (256, 128, 64, 32, 16), "decoder_attention_type": None, "in_channels": 3, "classes": 1, "activation": None, "aux_params": None } ``` ## Model metrics ```json [ { "test_per_image_iou": 0.7155967354774475, "test_dataset_iou": 0.6878355741500854 } ] ``` ## Dataset Dataset name: water-meter ## 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)

Source context: 7 downloads · 0 likes · Pipeline image-segmentation · Library segmentation-models-pytorch · Repo lanicc/water-meter-segmentation