Upload folder using huggingface_hub
Browse files- .gitattributes +35 -35
- README.md +177 -1
- config.json +54 -0
- disease_classification_metrics.png +0 -0
- model.safetensors +3 -0
- preprocessor_config.json +22 -0
.gitattributes
CHANGED
|
@@ -1,35 +1,35 @@
|
|
| 1 |
-
*.7z filter=lfs diff=lfs merge=lfs -text
|
| 2 |
-
*.arrow filter=lfs diff=lfs merge=lfs -text
|
| 3 |
-
*.bin filter=lfs diff=lfs merge=lfs -text
|
| 4 |
-
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
| 5 |
-
*.ckpt filter=lfs diff=lfs merge=lfs -text
|
| 6 |
-
*.ftz filter=lfs diff=lfs merge=lfs -text
|
| 7 |
-
*.gz filter=lfs diff=lfs merge=lfs -text
|
| 8 |
-
*.h5 filter=lfs diff=lfs merge=lfs -text
|
| 9 |
-
*.joblib filter=lfs diff=lfs merge=lfs -text
|
| 10 |
-
*.lfs.* filter=lfs diff=lfs merge=lfs -text
|
| 11 |
-
*.mlmodel filter=lfs diff=lfs merge=lfs -text
|
| 12 |
-
*.model filter=lfs diff=lfs merge=lfs -text
|
| 13 |
-
*.msgpack filter=lfs diff=lfs merge=lfs -text
|
| 14 |
-
*.npy filter=lfs diff=lfs merge=lfs -text
|
| 15 |
-
*.npz filter=lfs diff=lfs merge=lfs -text
|
| 16 |
-
*.onnx filter=lfs diff=lfs merge=lfs -text
|
| 17 |
-
*.ot filter=lfs diff=lfs merge=lfs -text
|
| 18 |
-
*.parquet filter=lfs diff=lfs merge=lfs -text
|
| 19 |
-
*.pb filter=lfs diff=lfs merge=lfs -text
|
| 20 |
-
*.pickle filter=lfs diff=lfs merge=lfs -text
|
| 21 |
-
*.pkl filter=lfs diff=lfs merge=lfs -text
|
| 22 |
-
*.pt filter=lfs diff=lfs merge=lfs -text
|
| 23 |
-
*.pth filter=lfs diff=lfs merge=lfs -text
|
| 24 |
-
*.rar filter=lfs diff=lfs merge=lfs -text
|
| 25 |
-
*.safetensors filter=lfs diff=lfs merge=lfs -text
|
| 26 |
-
saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
| 27 |
-
*.tar.* filter=lfs diff=lfs merge=lfs -text
|
| 28 |
-
*.tar filter=lfs diff=lfs merge=lfs -text
|
| 29 |
-
*.tflite filter=lfs diff=lfs merge=lfs -text
|
| 30 |
-
*.tgz filter=lfs diff=lfs merge=lfs -text
|
| 31 |
-
*.wasm filter=lfs diff=lfs merge=lfs -text
|
| 32 |
-
*.xz filter=lfs diff=lfs merge=lfs -text
|
| 33 |
-
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
-
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
-
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
| 1 |
+
*.7z filter=lfs diff=lfs merge=lfs -text
|
| 2 |
+
*.arrow filter=lfs diff=lfs merge=lfs -text
|
| 3 |
+
*.bin filter=lfs diff=lfs merge=lfs -text
|
| 4 |
+
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
| 5 |
+
*.ckpt filter=lfs diff=lfs merge=lfs -text
|
| 6 |
+
*.ftz filter=lfs diff=lfs merge=lfs -text
|
| 7 |
+
*.gz filter=lfs diff=lfs merge=lfs -text
|
| 8 |
+
*.h5 filter=lfs diff=lfs merge=lfs -text
|
| 9 |
+
*.joblib filter=lfs diff=lfs merge=lfs -text
|
| 10 |
+
*.lfs.* filter=lfs diff=lfs merge=lfs -text
|
| 11 |
+
*.mlmodel filter=lfs diff=lfs merge=lfs -text
|
| 12 |
+
*.model filter=lfs diff=lfs merge=lfs -text
|
| 13 |
+
*.msgpack filter=lfs diff=lfs merge=lfs -text
|
| 14 |
+
*.npy filter=lfs diff=lfs merge=lfs -text
|
| 15 |
+
*.npz filter=lfs diff=lfs merge=lfs -text
|
| 16 |
+
*.onnx filter=lfs diff=lfs merge=lfs -text
|
| 17 |
+
*.ot filter=lfs diff=lfs merge=lfs -text
|
| 18 |
+
*.parquet filter=lfs diff=lfs merge=lfs -text
|
| 19 |
+
*.pb filter=lfs diff=lfs merge=lfs -text
|
| 20 |
+
*.pickle filter=lfs diff=lfs merge=lfs -text
|
| 21 |
+
*.pkl filter=lfs diff=lfs merge=lfs -text
|
| 22 |
+
*.pt filter=lfs diff=lfs merge=lfs -text
|
| 23 |
+
*.pth filter=lfs diff=lfs merge=lfs -text
|
| 24 |
+
*.rar filter=lfs diff=lfs merge=lfs -text
|
| 25 |
+
*.safetensors filter=lfs diff=lfs merge=lfs -text
|
| 26 |
+
saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
| 27 |
+
*.tar.* filter=lfs diff=lfs merge=lfs -text
|
| 28 |
+
*.tar filter=lfs diff=lfs merge=lfs -text
|
| 29 |
+
*.tflite filter=lfs diff=lfs merge=lfs -text
|
| 30 |
+
*.tgz filter=lfs diff=lfs merge=lfs -text
|
| 31 |
+
*.wasm filter=lfs diff=lfs merge=lfs -text
|
| 32 |
+
*.xz filter=lfs diff=lfs merge=lfs -text
|
| 33 |
+
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
+
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
+
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
README.md
CHANGED
|
@@ -1,3 +1,179 @@
|
|
| 1 |
---
|
| 2 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
+
base_model:
|
| 3 |
+
- WinKawaks/vit-tiny-patch16-224
|
| 4 |
+
library_name: transformers
|
| 5 |
+
license: mit
|
| 6 |
+
metrics:
|
| 7 |
+
- accuracy
|
| 8 |
+
pipeline_tag: image-classification
|
| 9 |
+
tags:
|
| 10 |
+
- vision transformer
|
| 11 |
+
- agriculture
|
| 12 |
+
- plant disease detection
|
| 13 |
+
- smart farming
|
| 14 |
+
- image classification
|
| 15 |
---
|
| 16 |
+
|
| 17 |
+
# Fiche Modèle pour le Transformateur de Détection de Maladies en Agriculture Intelligente
|
| 18 |
+
|
| 19 |
+
Ce modèle est un Vision Transformer (ViT) conçu pour identifier les maladies des plantes dans les cultures dans le cadre d'un système d'agriculture intelligente. Il a été entraîné sur un ensemble de données diversifié d'images de plantes, incluant différentes catégories de maladies affectant les cultures telles que le maïs, la pomme de terre, le riz et le blé. Le modèle vise à fournir aux agriculteurs et agronomes une détection de maladies en temps réel pour une meilleure gestion des cultures.
|
| 20 |
+
|
| 21 |
+
## Détails du Modèle
|
| 22 |
+
|
| 23 |
+
### Description du Modèle
|
| 24 |
+
|
| 25 |
+
Ce modèle Vision Transformer a été affiné pour classifier diverses maladies de plantes couramment trouvées dans les environnements agricoles. Le modèle peut classifier les maladies dans les cultures telles que le maïs, la pomme de terre, le riz et le blé, identifiant des maladies comme la rouille, le mildiou, les taches foliaires, et autres. L'objectif est de permettre l'agriculture de précision en aidant les agriculteurs à détecter les maladies précocement et prendre les mesures appropriées.
|
| 26 |
+
|
| 27 |
+
- **Développé par :** Cabrel KEPSEU
|
| 28 |
+
- **Type de modèle :** Vision Transformer (ViT)
|
| 29 |
+
- **Langues (NLP) :** N/A (Modèle de Vision par Ordinateur)
|
| 30 |
+
- **Licence :** Apache 2.0
|
| 31 |
+
- **Affiné à partir du modèle :** (WinKawaks/vit-tiny-patch16-224)[https://huggingface.co/WinKawaks/vit-tiny-patch16-224]
|
| 32 |
+
- **Entrée :** Images de cultures (format RGB)
|
| 33 |
+
- **Sortie :** Étiquettes de classification de maladies (catégories saines ou malades)
|
| 34 |
+
## Maladies détectées par le modèle
|
| 35 |
+
|
| 36 |
+
| Culture | Maladies Identifiées |
|
| 37 |
+
|---------|------------------------------|
|
| 38 |
+
| Maïs | Rouille Commune |
|
| 39 |
+
| Maïs | Tache Grise des Feuilles |
|
| 40 |
+
| Maïs | Sain |
|
| 41 |
+
| Maïs | Brûlure des Feuilles |
|
| 42 |
+
| - | Invalide |
|
| 43 |
+
| Pomme de terre | Mildiou Précoce |
|
| 44 |
+
| Pomme de terre | Saine |
|
| 45 |
+
| Pomme de terre | Mildiou Tardif |
|
| 46 |
+
| Riz | Tache Brune |
|
| 47 |
+
| Riz | Sain |
|
| 48 |
+
| Riz | Pyriculariose |
|
| 49 |
+
| Blé | Rouille Brune |
|
| 50 |
+
| Blé | Sain |
|
| 51 |
+
| Blé | Rouille Jaune |
|
| 52 |
+
|
| 53 |
+
|
| 54 |
+
|
| 55 |
+
## Utilisations
|
| 56 |
+
|
| 57 |
+
### Utilisation Directe
|
| 58 |
+
|
| 59 |
+
Ce modèle peut être utilisé directement pour classifier des images de cultures afin de détecter les maladies des plantes. Il est particulièrement utile pour l'agriculture de précision, permettant aux utilisateurs de surveiller la santé des cultures et prendre des interventions précoces basées sur la maladie détectée.
|
| 60 |
+
|
| 61 |
+
### Utilisation en Aval
|
| 62 |
+
|
| 63 |
+
Ce modèle peut être affiné sur d'autres ensembles de données agricoles pour des cultures ou régions spécifiques afin d'améliorer ses performances ou être intégré dans des systèmes d'agriculture de précision plus larges qui incluent d'autres fonctionnalités comme les prédictions météorologiques et le contrôle d'irrigation.
|
| 64 |
+
|
| 65 |
+
Peut être quantifié ou déployé en pleine précision sur des appareils de périphérie grâce à sa petite taille de paramètres sans compromettre la précision et l'exactitude.
|
| 66 |
+
### Utilisation Hors Portée
|
| 67 |
+
|
| 68 |
+
Ce modèle n'est pas conçu pour des tâches de classification d'images non agricoles ou pour des environnements avec des données insuffisantes ou très bruitées. L'utilisation abusive inclut l'utilisation du modèle dans des zones avec des conditions agricoles très différentes de celles sur lesquelles il a été entraîné.
|
| 69 |
+
|
| 70 |
+
## Biais, Risques et Limitations
|
| 71 |
+
|
| 72 |
+
- Le modèle peut présenter un biais envers les cultures et maladies présentes dans l'ensemble de données d'entraînement, conduisant à des performances moindres sur les maladies ou variétés de cultures non représentées.
|
| 73 |
+
- Les faux négatifs (échec à détecter une maladie) peuvent résulter en des dommages de culture non traités, tandis que les faux positifs pourraient conduire à des interventions inutiles.
|
| 74 |
+
|
| 75 |
+
### Recommandations
|
| 76 |
+
|
| 77 |
+
Les utilisateurs devraient évaluer le modèle sur leurs cultures spécifiques et conditions agricoles. Des mises à jour régulières et un réentraînement avec des données locales sont recommandés pour des performances optimales.
|
| 78 |
+
|
| 79 |
+
## Comment Commencer avec le Modèle
|
| 80 |
+
|
| 81 |
+
```python
|
| 82 |
+
from PIL import Image, UnidentifiedImageError
|
| 83 |
+
from transformers import ViTFeatureExtractor, ViTForImageClassification
|
| 84 |
+
feature_extractor = ViTFeatureExtractor.from_pretrained('cabrel09/crop_leaf_diseases_detector')
|
| 85 |
+
model = ViTForImageClassification.from_pretrained(
|
| 86 |
+
'cabrel09/crop_leaf_diseases_detector',
|
| 87 |
+
ignore_mismatched_sizes=True
|
| 88 |
+
)
|
| 89 |
+
image = Image.open('<chemin_image>')
|
| 90 |
+
inputs = feature_extractor(images=image, return_tensors="pt")
|
| 91 |
+
outputs = model(**inputs)
|
| 92 |
+
logits = outputs.logits
|
| 93 |
+
predicted_class_idx = logits.argmax(-1).item()
|
| 94 |
+
print("Classe prédite:", model.config.id2label[predicted_class_idx])
|
| 95 |
+
```
|
| 96 |
+
|
| 97 |
+
## Détails d'Entraînement
|
| 98 |
+
|
| 99 |
+
### Données d'Entraînement
|
| 100 |
+
|
| 101 |
+
Le modèle a été entraîné sur un ensemble de données contenant des images de diverses cultures avec des maladies étiquetées, incluant les catégories suivantes :
|
| 102 |
+
|
| 103 |
+
- **Maïs** : Rouille Commune, Tache Grise des Feuilles, Brûlure des Feuilles, Sain
|
| 104 |
+
- **Pomme de terre** : Mildiou Précoce, Mildiou Tardif, Saine
|
| 105 |
+
- **Riz** : Tache Brune, Hispa, Pyriculariose, Sain
|
| 106 |
+
- **Blé** : Rouille Brune, Rouille Jaune, Sain
|
| 107 |
+
|
| 108 |
+
L'ensemble de données inclut également des images capturées sous diverses conditions d'éclairage, à partir d'environnements contrôlés et non contrôlés et d'angles différents, pour simuler des scénarios agricoles du monde réel.
|
| 109 |
+
Nous avons utilisé des ensembles de données publiquement disponibles, ainsi que nos propres données privées.
|
| 110 |
+
### Procédure d'Entraînement
|
| 111 |
+
|
| 112 |
+
Le modèle a été affiné en utilisant une architecture vision transformer pré-entraînée sur l'ensemble de données ImageNet. L'ensemble de données a été prétraité en redimensionnant les images et normalisant les valeurs de pixels.
|
| 113 |
+
|
| 114 |
+
#### Hyperparamètres d'Entraînement
|
| 115 |
+
|
| 116 |
+
- **Taille de lot :** 32
|
| 117 |
+
- **Taux d'apprentissage :** 2e-5
|
| 118 |
+
- **Époques :** 4
|
| 119 |
+
- **Optimiseur :** AdamW
|
| 120 |
+
- **Précision :** fp16
|
| 121 |
+
|
| 122 |
+
### Évaluation
|
| 123 |
+

|
| 124 |
+
|
| 125 |
+
|
| 126 |
+
#### Données de Test, Facteurs et Métriques
|
| 127 |
+
|
| 128 |
+
Le modèle a été évalué en utilisant un ensemble de validation composé de 20% de l'ensemble de données original, avec les métriques suivantes :
|
| 129 |
+
|
| 130 |
+
- **Précision :** 98%
|
| 131 |
+
- **Précision (Precision) :** 97%
|
| 132 |
+
- **Rappel :** 97%
|
| 133 |
+
- **Score F1 :** 96%
|
| 134 |
+
|
| 135 |
+
## Impact Environnemental
|
| 136 |
+
|
| 137 |
+
Les émissions de carbone pendant l'entraînement du modèle peuvent être estimées en utilisant le [calculateur d'impact de l'apprentissage automatique](https://mlco2.github.io/impact#compute).
|
| 138 |
+
|
| 139 |
+
- **Type de matériel :** NVIDIA L40S
|
| 140 |
+
- **Heures utilisées :** 1 heure
|
| 141 |
+
- **Fournisseur cloud :** Lightning AI
|
| 142 |
+
|
| 143 |
+
## Spécifications Techniques
|
| 144 |
+
|
| 145 |
+
### Architecture et Objectif du Modèle
|
| 146 |
+
|
| 147 |
+
Le modèle utilise une architecture Vision Transformer pour apprendre les représentations d'images et les classifier en catégories de maladies. Son mécanisme d'auto-attention lui permet de capturer l'information contextuelle globale dans les images, le rendant adapté à la détection de maladies agricoles.
|
| 148 |
+
|
| 149 |
+
### Infrastructure de Calcul
|
| 150 |
+
|
| 151 |
+
#### Logiciel
|
| 152 |
+
|
| 153 |
+
- Python 3.9
|
| 154 |
+
- PyTorch 2.4.1+cu121
|
| 155 |
+
- pytorch_lightning
|
| 156 |
+
- Bibliothèque Transformers par Hugging Face
|
| 157 |
+
|
| 158 |
+
## Citation
|
| 159 |
+
|
| 160 |
+
Si vous utilisez ce modèle dans vos recherches ou applications, veuillez le citer comme suit :
|
| 161 |
+
|
| 162 |
+
**BibTeX :**
|
| 163 |
+
|
| 164 |
+
```
|
| 165 |
+
@misc{cabrel2025cropdiseases,
|
| 166 |
+
title={Détecteur de maladies des feuilles de culture},
|
| 167 |
+
author={Cabrel Kepseu},
|
| 168 |
+
year={2025},
|
| 169 |
+
publisher={Hugging Face},
|
| 170 |
+
}
|
| 171 |
+
```
|
| 172 |
+
|
| 173 |
+
**APA :**
|
| 174 |
+
|
| 175 |
+
Cabrel, K. (2025). Détecteur de maladies des feuilles de culture. Hugging Face.
|
| 176 |
+
|
| 177 |
+
## Contact de la Fiche Modèle
|
| 178 |
+
|
| 179 |
+
Pour plus d'informations, contactez : klccomputing2023@gmail.com
|
config.json
ADDED
|
@@ -0,0 +1,54 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_name_or_path": "WinKawaks/vit-tiny-patch16-224",
|
| 3 |
+
"architectures": [
|
| 4 |
+
"ViTForImageClassification"
|
| 5 |
+
],
|
| 6 |
+
"attention_probs_dropout_prob": 0.0,
|
| 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": "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": 768,
|
| 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": 3,
|
| 47 |
+
"num_channels": 3,
|
| 48 |
+
"num_hidden_layers": 12,
|
| 49 |
+
"patch_size": 16,
|
| 50 |
+
"problem_type": "single_label_classification",
|
| 51 |
+
"qkv_bias": true,
|
| 52 |
+
"torch_dtype": "float32",
|
| 53 |
+
"transformers_version": "4.44.2"
|
| 54 |
+
}
|
disease_classification_metrics.png
ADDED
|
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1f91c9032e290d145bec27d8b82439bc9ea189f4c72923740daf4113990314b2
|
| 3 |
+
size 22130420
|
preprocessor_config.json
ADDED
|
@@ -0,0 +1,22 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"do_normalize": true,
|
| 3 |
+
"do_rescale": true,
|
| 4 |
+
"do_resize": true,
|
| 5 |
+
"image_mean": [
|
| 6 |
+
0.5,
|
| 7 |
+
0.5,
|
| 8 |
+
0.5
|
| 9 |
+
],
|
| 10 |
+
"image_processor_type": "ViTImageProcessor",
|
| 11 |
+
"image_std": [
|
| 12 |
+
0.5,
|
| 13 |
+
0.5,
|
| 14 |
+
0.5
|
| 15 |
+
],
|
| 16 |
+
"resample": 2,
|
| 17 |
+
"rescale_factor": 0.00392156862745098,
|
| 18 |
+
"size": {
|
| 19 |
+
"height": 224,
|
| 20 |
+
"width": 224
|
| 21 |
+
}
|
| 22 |
+
}
|