dcheng334 commited on
Commit
ed959d2
1 Parent(s): c77665d

Push model using huggingface_hub.

Browse files
Files changed (3) hide show
  1. README.md +3 -3
  2. config.json +1 -1
  3. model.safetensors +2 -2
README.md CHANGED
@@ -27,7 +27,7 @@ model = smp.FPN.from_pretrained("calfin")
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,
@@ -46,8 +46,8 @@ model_init_params = {
46
  ```json
47
  [
48
  {
49
- "test_per_image_iou": 0.7856383919715881,
50
- "test_dataset_iou": 0.7771378755569458
51
  }
52
  ]
53
  ```
 
27
  ## Model init parameters
28
  ```python
29
  model_init_params = {
30
+ "encoder_name": "xception",
31
  "encoder_depth": 5,
32
  "encoder_weights": "imagenet",
33
  "decoder_pyramid_channels": 256,
 
46
  ```json
47
  [
48
  {
49
+ "test_per_image_iou": 0.6299741864204407,
50
+ "test_dataset_iou": 0.6309026479721069
51
  }
52
  ]
53
  ```
config.json CHANGED
@@ -8,7 +8,7 @@
8
  "decoder_pyramid_channels": 256,
9
  "decoder_segmentation_channels": 128,
10
  "encoder_depth": 5,
11
- "encoder_name": "resnet34",
12
  "encoder_weights": "imagenet",
13
  "in_channels": 3,
14
  "upsampling": 4
 
8
  "decoder_pyramid_channels": 256,
9
  "decoder_segmentation_channels": 128,
10
  "encoder_depth": 5,
11
+ "encoder_name": "xception",
12
  "encoder_weights": "imagenet",
13
  "in_channels": 3,
14
  "upsampling": 4
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:077705d5074108724eabd175fbfeacf55c5fc46441840f61bdc8126124e002e0
3
- size 92714324
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6053508db159f1e70299e5b6aa82b04ef27427abc7539710aac2cb44b5203932
3
+ size 93212444