qubvel-hf commited on
Commit
e1788aa
·
verified ·
1 Parent(s): 1a3a5e8

Upload folder using huggingface_hub

Browse files
Files changed (4) hide show
  1. README.md +3 -2
  2. config.json +4 -3
  3. model.safetensors +2 -2
  4. output-tensor.pth +1 -1
README.md CHANGED
@@ -32,11 +32,12 @@ model_init_params = {
32
  "encoder_name": "tu-resnet18",
33
  "encoder_depth": 5,
34
  "encoder_weights": "imagenet",
35
- "decoder_pyramid_channels": 256,
36
- "decoder_segmentation_channels": 64,
37
  "in_channels": 3,
38
  "classes": 1,
39
  "activation": None,
 
40
  "aux_params": None
41
  }
42
  ```
 
32
  "encoder_name": "tu-resnet18",
33
  "encoder_depth": 5,
34
  "encoder_weights": "imagenet",
35
+ "decoder_channels": 256,
36
+ "decoder_use_norm": "batchnorm",
37
  "in_channels": 3,
38
  "classes": 1,
39
  "activation": None,
40
+ "upsampling": 4,
41
  "aux_params": None
42
  }
43
  ```
config.json CHANGED
@@ -3,10 +3,11 @@
3
  "activation": null,
4
  "aux_params": null,
5
  "classes": 1,
6
- "decoder_pyramid_channels": 256,
7
- "decoder_segmentation_channels": 64,
8
  "encoder_depth": 5,
9
  "encoder_name": "tu-resnet18",
10
  "encoder_weights": "imagenet",
11
- "in_channels": 3
 
12
  }
 
3
  "activation": null,
4
  "aux_params": null,
5
  "classes": 1,
6
+ "decoder_channels": 256,
7
+ "decoder_use_norm": "batchnorm",
8
  "encoder_depth": 5,
9
  "encoder_name": "tu-resnet18",
10
  "encoder_weights": "imagenet",
11
+ "in_channels": 3,
12
+ "upsampling": 4
13
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:66ab4bc4f5fce64c40ecaf484a61d983537f7cfb667ec96d2aa5a23ee2814286
3
- size 50371828
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ed1fb6ceffe7f8413f625b85ab549bc1b7bd4b53568ff78473f5f4fea662b382
3
+ size 78050444
output-tensor.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5272f51b3042ccba3600e2ace7114858da13e092c017fe025d3430c96401ef39
3
  size 263354
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:de83edc850a6874f49b76c61dbc3cf195363b07431b10ce1830627c3157f4bc5
3
  size 263354