Hugging Face's logo

wambugu71
/
crop_leaf_diseases_vit

Image Classification
Transformers
Safetensors
vit
vision transformer
agriculture
plant disease detection
smart farming
image classification
Inference Endpoints
Model card Files Community
Kennedy wambugu commited on
Commit
bb1d283
·
verified ·
1 Parent(s): eeb2de9

Upload ViTForImageClassification

Browse files
Files changed (2) hide show
  1. config.json +24 -26
  2. model.safetensors +2 -2
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "WinKawaks/vit-tiny-patch16-224",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
@@ -7,45 +7,43 @@
7
  "encoder_stride": 16,
8
  "hidden_act": "gelu",
9
  "hidden_dropout_prob": 0.0,
10
- "hidden_size": 192,
11
  "id2label": {
12
  "0": "Corn___Common_Rust",
13
  "1": "Corn___Gray_Leaf_Spot",
14
- "10": "Rice___Leaf_Blast",
15
- "11": "Wheat___Brown_Rust",
16
- "12": "Wheat___Healthy",
17
- "13": "Wheat___Yellow_Rust",
18
  "2": "Corn___Healthy",
19
- "3": "Corn___Leaf_Blight",
20
- "4": "Invalid",
21
- "5": "Potato___Early_Blight",
22
- "6": "Potato___Healthy",
23
- "7": "Potato___Late_Blight",
24
- "8": "Rice___Brown_Spot",
25
- "9": "Rice___Healthy"
26
  },
27
  "image_size": 224,
28
  "initializer_range": 0.02,
29
- "intermediate_size": 768,
30
  "label2id": {
31
  "Corn___Common_Rust": "0",
32
  "Corn___Gray_Leaf_Spot": "1",
33
  "Corn___Healthy": "2",
34
- "Corn___Leaf_Blight": "3",
35
- "Invalid": "4",
36
- "Potato___Early_Blight": "5",
37
- "Potato___Healthy": "6",
38
- "Potato___Late_Blight": "7",
39
- "Rice___Brown_Spot": "8",
40
- "Rice___Healthy": "9",
41
- "Rice___Leaf_Blast": "10",
42
- "Wheat___Brown_Rust": "11",
43
- "Wheat___Healthy": "12",
44
- "Wheat___Yellow_Rust": "13"
45
  },
46
  "layer_norm_eps": 1e-12,
47
  "model_type": "vit",
48
- "num_attention_heads": 3,
49
  "num_channels": 3,
50
  "num_hidden_layers": 12,
51
  "patch_size": 16,
 
1
  {
2
+ "_name_or_path": "google/vit-base-patch16-224",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
 
7
  "encoder_stride": 16,
8
  "hidden_act": "gelu",
9
  "hidden_dropout_prob": 0.0,
10
+ "hidden_size": 768,
11
  "id2label": {
12
  "0": "Corn___Common_Rust",
13
  "1": "Corn___Gray_Leaf_Spot",
14
+ "10": "Wheat___Brown_Rust",
15
+ "11": "Wheat___Healthy",
16
+ "12": "Wheat___Yellow_Rust",
 
17
  "2": "Corn___Healthy",
18
+ "3": "Invalid",
19
+ "4": "Potato___Early_Blight",
20
+ "5": "Potato___Healthy",
21
+ "6": "Potato___Late_Blight",
22
+ "7": "Rice___Brown_Spot",
23
+ "8": "Rice___Healthy",
24
+ "9": "Rice___Leaf_Blast"
25
  },
26
  "image_size": 224,
27
  "initializer_range": 0.02,
28
+ "intermediate_size": 3072,
29
  "label2id": {
30
  "Corn___Common_Rust": "0",
31
  "Corn___Gray_Leaf_Spot": "1",
32
  "Corn___Healthy": "2",
33
+ "Invalid": "3",
34
+ "Potato___Early_Blight": "4",
35
+ "Potato___Healthy": "5",
36
+ "Potato___Late_Blight": "6",
37
+ "Rice___Brown_Spot": "7",
38
+ "Rice___Healthy": "8",
39
+ "Rice___Leaf_Blast": "9",
40
+ "Wheat___Brown_Rust": "10",
41
+ "Wheat___Healthy": "11",
42
+ "Wheat___Yellow_Rust": "12"
 
43
  },
44
  "layer_norm_eps": 1e-12,
45
  "model_type": "vit",
46
+ "num_attention_heads": 12,
47
  "num_channels": 3,
48
  "num_hidden_layers": 12,
49
  "patch_size": 16,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:faa3a311176efe310e855f564298826276e328db73e26aae9458e414fccf877a
3
- size 22131192
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:177ca2ffc86b628d58b456194e5362d72d7c96945e5ddec24e21bbae3ed62d25
3
+ size 343257812