qubvel-hf HF staff commited on
Commit
6916387
1 Parent(s): 1b36419

Push model using huggingface_hub.

Browse files
Files changed (3) hide show
  1. README.md +62 -0
  2. config.json +14 -0
  3. model.safetensors +3 -0
README.md ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ library_name: segmentation-models-pytorch
4
+ tags:
5
+ - semantic-segmentation
6
+ - pytorch
7
+ - segmentation-models-pytorch
8
+ languages:
9
+ - python
10
+ pipeline_tag: image-segmentation
11
+ ---
12
+ # FPN Model Card
13
+
14
+ Table of Contents:
15
+ - [Load trained model](#load-trained-model)
16
+ - [Model init parameters](#model-init-parameters)
17
+ - [Model metrics](#model-metrics)
18
+ - [Dataset](#dataset)
19
+
20
+ ## Load trained model
21
+ ```python
22
+ import segmentation_models_pytorch as smp
23
+
24
+ model = smp.FPN.from_pretrained("oxford-pet-segmentation")
25
+ ```
26
+
27
+ ## Model init parameters
28
+ ```python
29
+ model_init_params = {
30
+ "encoder_name": "resnet34",
31
+ "encoder_depth": 5,
32
+ "encoder_weights": "imagenet",
33
+ "decoder_pyramid_channels": 256,
34
+ "decoder_segmentation_channels": 128,
35
+ "decoder_merge_policy": "add",
36
+ "decoder_dropout": 0.2,
37
+ "in_channels": 3,
38
+ "classes": 1,
39
+ "activation": None,
40
+ "upsampling": 4,
41
+ "aux_params": None
42
+ }
43
+ ```
44
+
45
+ ## Model metrics
46
+ ```json
47
+ [
48
+ {
49
+ "test_per_image_iou": 0.9058760404586792,
50
+ "test_dataset_iou": 0.9127062559127808
51
+ }
52
+ ]
53
+ ```
54
+
55
+ ## Dataset
56
+ Dataset name: Oxford Pet
57
+
58
+ ## More Information
59
+ - Library: https://github.com/qubvel/segmentation_models.pytorch
60
+ - Docs: https://smp.readthedocs.io/en/latest/
61
+
62
+ This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin)
config.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation": null,
3
+ "aux_params": null,
4
+ "classes": 1,
5
+ "decoder_dropout": 0.2,
6
+ "decoder_merge_policy": "add",
7
+ "decoder_pyramid_channels": 256,
8
+ "decoder_segmentation_channels": 128,
9
+ "encoder_depth": 5,
10
+ "encoder_name": "resnet34",
11
+ "encoder_weights": "imagenet",
12
+ "in_channels": 3,
13
+ "upsampling": 4
14
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9ad708c7f6d57d6ac0340296e9ee49f0519c19f2f51b78bdcbd3db79433c2565
3
+ size 92714324