Image Classification
Transformers
PyTorch
TensorFlow
TensorBoard
Safetensors
timm_wrapper
vision
Generated from Trainer
Instructions to use amyeroberts/vit-base-beans with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use amyeroberts/vit-base-beans with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="amyeroberts/vit-base-beans") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")# Load model directly from transformers import AutoImageProcessor, AutoModelForImageClassification processor = AutoImageProcessor.from_pretrained("amyeroberts/vit-base-beans") model = AutoModelForImageClassification.from_pretrained("amyeroberts/vit-base-beans", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Model save
Browse files
README.md
CHANGED
|
@@ -1,45 +1,25 @@
|
|
| 1 |
---
|
|
|
|
| 2 |
license: apache-2.0
|
|
|
|
| 3 |
tags:
|
| 4 |
-
- image-classification
|
| 5 |
-
- vision
|
| 6 |
- generated_from_trainer
|
| 7 |
-
datasets:
|
| 8 |
-
- AI-Lab-Makerere/beans
|
| 9 |
metrics:
|
| 10 |
- accuracy
|
| 11 |
-
base_model: google/vit-base-patch16-224-in21k
|
| 12 |
model-index:
|
| 13 |
- name: vit-base-beans
|
| 14 |
-
results:
|
| 15 |
-
- task:
|
| 16 |
-
type: image-classification
|
| 17 |
-
name: Image Classification
|
| 18 |
-
dataset:
|
| 19 |
-
name: beans
|
| 20 |
-
type: beans
|
| 21 |
-
config: default
|
| 22 |
-
split: validation
|
| 23 |
-
args: default
|
| 24 |
-
metrics:
|
| 25 |
-
- type: accuracy
|
| 26 |
-
value: 0.9849624060150376
|
| 27 |
-
name: Accuracy
|
| 28 |
---
|
| 29 |
|
| 30 |
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
| 31 |
should probably proofread and complete it, then remove this comment. -->
|
| 32 |
|
| 33 |
-
# THIS IS A TEST REPO FOR DEBUGGING!
|
| 34 |
-
|
| 35 |
-
This repo is here as a result of playing with and debugging training scripts and push to hub features. As such, the TesnorFlow and PyTorch models will be out of sync and different weights may be push at any time, including pushing models with very low performance.
|
| 36 |
-
|
| 37 |
# vit-base-beans
|
| 38 |
|
| 39 |
-
This model is a fine-tuned version of [
|
| 40 |
It achieves the following results on the evaluation set:
|
| 41 |
-
- Loss:
|
| 42 |
-
- Accuracy: 0.
|
| 43 |
|
| 44 |
## Model description
|
| 45 |
|
|
@@ -70,16 +50,16 @@ The following hyperparameters were used during training:
|
|
| 70 |
|
| 71 |
| Training Loss | Epoch | Step | Validation Loss | Accuracy |
|
| 72 |
|:-------------:|:-----:|:----:|:---------------:|:--------:|
|
| 73 |
-
|
|
| 74 |
-
|
|
| 75 |
-
|
|
| 76 |
-
|
|
| 77 |
-
|
|
| 78 |
|
| 79 |
|
| 80 |
### Framework versions
|
| 81 |
|
| 82 |
-
- Transformers 4.
|
| 83 |
-
- Pytorch
|
| 84 |
-
- Datasets 2.
|
| 85 |
-
- Tokenizers 0.
|
|
|
|
| 1 |
---
|
| 2 |
+
library_name: transformers
|
| 3 |
license: apache-2.0
|
| 4 |
+
base_model: timm/resnet18.a1_in1k
|
| 5 |
tags:
|
|
|
|
|
|
|
| 6 |
- generated_from_trainer
|
|
|
|
|
|
|
| 7 |
metrics:
|
| 8 |
- accuracy
|
|
|
|
| 9 |
model-index:
|
| 10 |
- name: vit-base-beans
|
| 11 |
+
results: []
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
---
|
| 13 |
|
| 14 |
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
| 15 |
should probably proofread and complete it, then remove this comment. -->
|
| 16 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 17 |
# vit-base-beans
|
| 18 |
|
| 19 |
+
This model is a fine-tuned version of [timm/resnet18.a1_in1k](https://huggingface.co/timm/resnet18.a1_in1k) on an unknown dataset.
|
| 20 |
It achieves the following results on the evaluation set:
|
| 21 |
+
- Loss: 1.0324
|
| 22 |
+
- Accuracy: 0.6917
|
| 23 |
|
| 24 |
## Model description
|
| 25 |
|
|
|
|
| 50 |
|
| 51 |
| Training Loss | Epoch | Step | Validation Loss | Accuracy |
|
| 52 |
|:-------------:|:-----:|:----:|:---------------:|:--------:|
|
| 53 |
+
| 1.0884 | 1.0 | 130 | 1.0903 | 0.4060 |
|
| 54 |
+
| 1.0721 | 2.0 | 260 | 1.0681 | 0.5188 |
|
| 55 |
+
| 1.0623 | 3.0 | 390 | 1.0460 | 0.6391 |
|
| 56 |
+
| 1.052 | 4.0 | 520 | 1.0410 | 0.6165 |
|
| 57 |
+
| 1.0519 | 5.0 | 650 | 1.0324 | 0.6917 |
|
| 58 |
|
| 59 |
|
| 60 |
### Framework versions
|
| 61 |
|
| 62 |
+
- Transformers 4.46.0.dev0
|
| 63 |
+
- Pytorch 2.4.0
|
| 64 |
+
- Datasets 2.15.1.dev0
|
| 65 |
+
- Tokenizers 0.20.0
|
emissions.csv
CHANGED
|
@@ -1,2 +1,2 @@
|
|
| 1 |
timestamp,experiment_id,project_name,duration,emissions,energy_consumed,country_name,country_iso_code,region,on_cloud,cloud_provider,cloud_region
|
| 2 |
-
|
|
|
|
| 1 |
timestamp,experiment_id,project_name,duration,emissions,energy_consumed,country_name,country_iso_code,region,on_cloud,cloud_provider,cloud_region
|
| 2 |
+
2024-09-30T20:12:46,a1b98ce0-5999-43a7-ae1d-c76c57ba4514,codecarbon,89.43079400062561,0.0007095535156443428,0.0010557118809902488,United Kingdom,GBR,england,N,,
|
runs/Sep30_20-11-09_Amys-MBP-2.communityfibre.co.uk/events.out.tfevents.1727723476.Amys-MBP-2.communityfibre.co.uk.11627.0
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2175acb01da297e3533348e246f583eae4f510adffac1fd4eb3b4f49a998cbaf
|
| 3 |
+
size 21166
|