wambugu71
/

crop_leaf_diseases_vit

Image Classification

vision transformer

plant disease detection

image classification

Inference Endpoints

Model card Files Files and versions Community

Kennedy wambugu commited on Oct 11, 2024

Commit

311a821

·

verified ·

1 Parent(s): bb1d283

Upload ViTForImageClassification

Files changed (2) hide show

config.json +4 -4
model.safetensors +2 -2

config.json CHANGED Viewed

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "google/vit-base-patch16-224",
   "architectures": [
     "ViTForImageClassification"
   ],
@@ -7,7 +7,7 @@
   "encoder_stride": 16,
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.0,
-  "hidden_size": 768,
   "id2label": {
     "0": "Corn___Common_Rust",
     "1": "Corn___Gray_Leaf_Spot",
@@ -25,7 +25,7 @@
   },
   "image_size": 224,
   "initializer_range": 0.02,
-  "intermediate_size": 3072,
   "label2id": {
     "Corn___Common_Rust": "0",
     "Corn___Gray_Leaf_Spot": "1",
@@ -43,7 +43,7 @@
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",
-  "num_attention_heads": 12,
   "num_channels": 3,
   "num_hidden_layers": 12,
   "patch_size": 16,

 {
+  "_name_or_path": "WinKawaks/vit-tiny-patch16-224",
   "architectures": [
     "ViTForImageClassification"
   ],
   "encoder_stride": 16,
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.0,
+  "hidden_size": 192,
   "id2label": {
     "0": "Corn___Common_Rust",
     "1": "Corn___Gray_Leaf_Spot",
   },
   "image_size": 224,
   "initializer_range": 0.02,
+  "intermediate_size": 768,
   "label2id": {
     "Corn___Common_Rust": "0",
     "Corn___Gray_Leaf_Spot": "1",
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",
+  "num_attention_heads": 3,
   "num_channels": 3,
   "num_hidden_layers": 12,
   "patch_size": 16,

model.safetensors CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:177ca2ffc86b628d58b456194e5362d72d7c96945e5ddec24e21bbae3ed62d25
-size 343257812

 version https://git-lfs.github.com/spec/v1
+oid sha256:1f91c9032e290d145bec27d8b82439bc9ea189f4c72923740daf4113990314b2
+size 22130420