Straueri commited on
Commit
cf89e25
·
verified ·
1 Parent(s): 72b0533

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +69 -0
README.md ADDED
@@ -0,0 +1,69 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ base_model: torchvision/resnet50
5
+ tags:
6
+ - image-classification
7
+ - computer-vision
8
+ - pytorch
9
+ - reptiles
10
+ - amphibians
11
+ metrics:
12
+ - accuracy
13
+ model-index:
14
+ - name: resnet50-reptile-amphibian-classifier
15
+ results: []
16
+ ---
17
+
18
+ <!-- Diese Model Card wurde auf Basis des Trainings-Notebooks automatisch erstellt. Bitte prüfen, ergänzen und diesen Kommentar ggf. entfernen. -->
19
+
20
+ # resnet50-reptile-amphibian-classifier
21
+
22
+ Dieses Modell ist eine feinjustierte Version von ResNet-50 zur Bildklassifikation verschiedener Reptilien- und Amphibienarten. Es wurde auf einem Bilddatensatz trainiert, der von der Plattform Kaggle stammt und Arten wie Frösche, Eidechsen, Schlangen usw. umfasst.
23
+
24
+ ## Modellbeschreibung
25
+
26
+ Das Modell basiert auf der `resnet50`-Architektur aus `torchvision.models`, wobei die finale Klassifikationsschicht (`fc`) an die Anzahl der Klassen im Datensatz angepasst wurde. Es wurde mithilfe von Data Augmentation und der `Adam`-Optimierung trainiert.
27
+
28
+ ## Anwendungsgebiete & Einschränkungen
29
+
30
+ **Geeignet für:**
31
+ - Klassifikation von Bildern mit Reptilien und Amphibien
32
+ - Zoologische oder biologische Bildanalysen
33
+
34
+ **Nicht geeignet für:**
35
+ - Bilder mit schlechter Qualität oder aus anderen Domänen
36
+ - Klassifikation von nicht-reptilienartigen Tieren
37
+
38
+ ## Trainings- und Evaluierungsdaten
39
+
40
+ Der Datensatz besteht aus Bildern, die in die Klassenverzeichnisse unter `data/train`, `data/val` und `data/test` aufgeteilt wurden. Er wurde automatisch in Trainings-, Validierungs- und Testsets im Verhältnis 70/15/15 aufgeteilt.
41
+
42
+ ## Trainingsprozedur
43
+
44
+ ### Trainings-Hyperparameter
45
+
46
+ - learning_rate: 0.001
47
+ - train_batch_size: 32
48
+ - eval_batch_size: 32
49
+ - seed: 42
50
+ - optimizer: Adam
51
+ - lr_scheduler_type: ReduceLROnPlateau
52
+ - num_epochs: 15
53
+
54
+ ### Trainingsergebnisse
55
+
56
+ | Training Loss | Validation Loss | Validation Accuracy |
57
+ |:-------------:|:---------------:|:-------------------:|
58
+ | ≈0.13 (final) | ≈0.18 (final) | ≈0.94–0.95 |
59
+
60
+ (Genauere Werte sind im Notebook-Ausgang sichtbar)
61
+
62
+ ## Framework-Versionen
63
+
64
+ - Transformers: nicht verwendet
65
+ - PyTorch: ≥ 2.0
66
+ - Torchvision: ≥ 0.15
67
+ - Scikit-learn: zur Auswertung verwendet
68
+ - Gradio (optional): für Deployment
69
+