Hugging Face's logo

jwinn03
/
segment-cattle

Safetensors
segformer
Model card Files
xet
Community
jwinn03 commited on
Commit
09dde60
·
verified ·
1 Parent(s): 1fb9a27

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. config.json +77 -78
  2. model.safetensors +2 -2
config.json CHANGED
@@ -1,78 +1,77 @@
1
- {
2
- "_name_or_path": "nvidia/mit-b0",
3
- "architectures": [
4
- "SegformerForSemanticSegmentation"
5
- ],
6
- "attention_probs_dropout_prob": 0.0,
7
- "classifier_dropout_prob": 0.1,
8
- "decoder_hidden_size": 256,
9
- "depths": [
10
- 2,
11
- 2,
12
- 2,
13
- 2
14
- ],
15
- "downsampling_rates": [
16
- 1,
17
- 4,
18
- 8,
19
- 16
20
- ],
21
- "drop_path_rate": 0.1,
22
- "hidden_act": "gelu",
23
- "hidden_dropout_prob": 0.0,
24
- "hidden_sizes": [
25
- 32,
26
- 64,
27
- 160,
28
- 256
29
- ],
30
- "id2label": {
31
- "0": "cattle",
32
- "1": "background"
33
- },
34
- "image_size": 224,
35
- "initializer_range": 0.02,
36
- "label2id": {
37
- "background": 1,
38
- "cattle": 0
39
- },
40
- "layer_norm_eps": 1e-06,
41
- "mlp_ratios": [
42
- 4,
43
- 4,
44
- 4,
45
- 4
46
- ],
47
- "model_type": "segformer",
48
- "num_attention_heads": [
49
- 1,
50
- 2,
51
- 5,
52
- 8
53
- ],
54
- "num_channels": 3,
55
- "num_encoder_blocks": 4,
56
- "patch_sizes": [
57
- 7,
58
- 3,
59
- 3,
60
- 3
61
- ],
62
- "reshape_last_stage": true,
63
- "semantic_loss_ignore_index": 255,
64
- "sr_ratios": [
65
- 8,
66
- 4,
67
- 2,
68
- 1
69
- ],
70
- "strides": [
71
- 4,
72
- 2,
73
- 2,
74
- 2
75
- ],
76
- "torch_dtype": "float32",
77
- "transformers_version": "4.46.2"
78
- }
 
1
+ {
2
+ "architectures": [
3
+ "SegformerForSemanticSegmentation"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.0,
6
+ "classifier_dropout_prob": 0.1,
7
+ "decoder_hidden_size": 256,
8
+ "depths": [
9
+ 2,
10
+ 2,
11
+ 2,
12
+ 2
13
+ ],
14
+ "downsampling_rates": [
15
+ 1,
16
+ 4,
17
+ 8,
18
+ 16
19
+ ],
20
+ "drop_path_rate": 0.1,
21
+ "hidden_act": "gelu",
22
+ "hidden_dropout_prob": 0.0,
23
+ "hidden_sizes": [
24
+ 64,
25
+ 128,
26
+ 320,
27
+ 512
28
+ ],
29
+ "id2label": {
30
+ "0": "cattle",
31
+ "1": "background"
32
+ },
33
+ "image_size": 224,
34
+ "initializer_range": 0.02,
35
+ "label2id": {
36
+ "background": 1,
37
+ "cattle": 0
38
+ },
39
+ "layer_norm_eps": 1e-06,
40
+ "mlp_ratios": [
41
+ 4,
42
+ 4,
43
+ 4,
44
+ 4
45
+ ],
46
+ "model_type": "segformer",
47
+ "num_attention_heads": [
48
+ 1,
49
+ 2,
50
+ 5,
51
+ 8
52
+ ],
53
+ "num_channels": 3,
54
+ "num_encoder_blocks": 4,
55
+ "patch_sizes": [
56
+ 7,
57
+ 3,
58
+ 3,
59
+ 3
60
+ ],
61
+ "reshape_last_stage": true,
62
+ "semantic_loss_ignore_index": 255,
63
+ "sr_ratios": [
64
+ 8,
65
+ 4,
66
+ 2,
67
+ 1
68
+ ],
69
+ "strides": [
70
+ 4,
71
+ 2,
72
+ 2,
73
+ 2
74
+ ],
75
+ "torch_dtype": "float32",
76
+ "transformers_version": "4.51.3"
77
+ }
 
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4606d05763d1cbbe85394901fa530efc650b70fb79b1c30c389709443ea10643
3
- size 14884776
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4443a58a657a1fcf05c1642124c91ea23a57679932e19a3e425bf661f639bae0
3
+ size 54737376