Hugging Face's logo

SodaXII
/
dinov2-base_rice-leaf-disease-augmented-v4_tl

Image Classification
Transformers
TensorBoard
Safetensors
dinov2
Generated from Trainer
Model card Files Metrics Community
SodaXII commited on
Commit
f51f95e
·
verified ·
1 Parent(s): 559d542

Model save

Browse files
Files changed (8) hide show
  1. README.md +92 -0
  2. config.json +69 -0
  3. logs/events.out.tfevents.1741874217.08c8ff51aa91.308.0 +3 -0
  4. logs/events.out.tfevents.1741879037.ba36c15cdafa.233.0 +3 -0
  5. model.safetensors +3 -0
  6. preprocessor_config.json +27 -0
  7. training_args.bin +3 -0
  8. training_metrics.xlsx +0 -0
README.md ADDED
@@ -0,0 +1,92 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ base_model: facebook/dinov2-base
5
+ tags:
6
+ - generated_from_trainer
7
+ metrics:
8
+ - accuracy
9
+ model-index:
10
+ - name: dinov2-base_rice-leaf-disease-augmented-v4_tl
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
+ # dinov2-base_rice-leaf-disease-augmented-v4_tl
18
+
19
+ This model is a fine-tuned version of [facebook/dinov2-base](https://huggingface.co/facebook/dinov2-base) on an unknown dataset.
20
+ It achieves the following results on the evaluation set:
21
+ - Loss: 0.2961
22
+ - Accuracy: 0.9060
23
+
24
+ ## Model description
25
+
26
+ More information needed
27
+
28
+ ## Intended uses & limitations
29
+
30
+ More information needed
31
+
32
+ ## Training and evaluation data
33
+
34
+ More information needed
35
+
36
+ ## Training procedure
37
+
38
+ ### Training hyperparameters
39
+
40
+ The following hyperparameters were used during training:
41
+ - learning_rate: 0.0003
42
+ - train_batch_size: 64
43
+ - eval_batch_size: 64
44
+ - seed: 42
45
+ - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
46
+ - lr_scheduler_type: cosine_with_restarts
47
+ - lr_scheduler_warmup_steps: 256
48
+ - num_epochs: 15
49
+ - mixed_precision_training: Native AMP
50
+
51
+ ### Training results
52
+
53
+ | Training Loss | Epoch | Step | Accuracy | Validation Loss |
54
+ |:-------------:|:-----:|:----:|:--------:|:---------------:|
55
+ | 2.1181 | 0.5 | 64 | 0.3658 | 1.7085 |
56
+ | 1.3815 | 1.0 | 128 | 0.7047 | 1.0554 |
57
+ | 0.8981 | 1.5 | 192 | 0.7785 | 0.7380 |
58
+ | 0.6515 | 2.0 | 256 | 0.7987 | 0.5983 |
59
+ | 0.5233 | 2.5 | 320 | 0.8255 | 0.5212 |
60
+ | 0.4535 | 3.0 | 384 | 0.8356 | 0.4803 |
61
+ | 0.3959 | 3.5 | 448 | 0.8557 | 0.4428 |
62
+ | 0.3614 | 4.0 | 512 | 0.8725 | 0.4040 |
63
+ | 0.3391 | 4.5 | 576 | 0.8691 | 0.4082 |
64
+ | 0.318 | 5.0 | 640 | 0.3749 | 0.8859 |
65
+ | 0.3134 | 5.5 | 704 | 0.3922 | 0.8758 |
66
+ | 0.3025 | 6.0 | 768 | 0.3802 | 0.8826 |
67
+ | 0.3098 | 6.5 | 832 | 0.3856 | 0.8725 |
68
+ | 0.2986 | 7.0 | 896 | 0.3666 | 0.8859 |
69
+ | 0.2856 | 7.5 | 960 | 0.3525 | 0.8826 |
70
+ | 0.2549 | 8.0 | 1024 | 0.3327 | 0.8725 |
71
+ | 0.24 | 8.5 | 1088 | 0.3428 | 0.9027 |
72
+ | 0.2422 | 9.0 | 1152 | 0.3304 | 0.8993 |
73
+ | 0.2178 | 9.5 | 1216 | 0.3238 | 0.9027 |
74
+ | 0.2297 | 10.0 | 1280 | 0.3273 | 0.9027 |
75
+ | 0.2176 | 10.5 | 1344 | 0.3285 | 0.9027 |
76
+ | 0.2236 | 11.0 | 1408 | 0.3411 | 0.8792 |
77
+ | 0.2198 | 11.5 | 1472 | 0.3342 | 0.8893 |
78
+ | 0.2056 | 12.0 | 1536 | 0.3108 | 0.9060 |
79
+ | 0.1915 | 12.5 | 1600 | 0.2995 | 0.8993 |
80
+ | 0.1911 | 13.0 | 1664 | 0.3058 | 0.9027 |
81
+ | 0.1787 | 13.5 | 1728 | 0.2928 | 0.9128 |
82
+ | 0.1776 | 14.0 | 1792 | 0.2979 | 0.9060 |
83
+ | 0.1724 | 14.5 | 1856 | 0.2966 | 0.9060 |
84
+ | 0.168 | 15.0 | 1920 | 0.2961 | 0.9060 |
85
+
86
+
87
+ ### Framework versions
88
+
89
+ - Transformers 4.48.3
90
+ - Pytorch 2.5.1+cu124
91
+ - Datasets 3.3.2
92
+ - Tokenizers 0.21.0
config.json ADDED
@@ -0,0 +1,69 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "facebook/dinov2-base",
3
+ "apply_layernorm": true,
4
+ "architectures": [
5
+ "Dinov2ForImageClassification"
6
+ ],
7
+ "attention_probs_dropout_prob": 0.0,
8
+ "drop_path_rate": 0.0,
9
+ "hidden_act": "gelu",
10
+ "hidden_dropout_prob": 0.0,
11
+ "hidden_size": 768,
12
+ "id2label": {
13
+ "0": "Bacterial Leaf Blight",
14
+ "1": "Brown Spot",
15
+ "2": "Healthy Rice Leaf",
16
+ "3": "Leaf Blast",
17
+ "4": "Leaf scald",
18
+ "5": "Narrow Brown Leaf Spot",
19
+ "6": "Rice Hispa",
20
+ "7": "Sheath Blight"
21
+ },
22
+ "image_size": 518,
23
+ "initializer_range": 0.02,
24
+ "label2id": {
25
+ "Bacterial Leaf Blight": 0,
26
+ "Brown Spot": 1,
27
+ "Healthy Rice Leaf": 2,
28
+ "Leaf Blast": 3,
29
+ "Leaf scald": 4,
30
+ "Narrow Brown Leaf Spot": 5,
31
+ "Rice Hispa": 6,
32
+ "Sheath Blight": 7
33
+ },
34
+ "layer_norm_eps": 1e-06,
35
+ "layerscale_value": 1.0,
36
+ "mlp_ratio": 4,
37
+ "model_type": "dinov2",
38
+ "num_attention_heads": 12,
39
+ "num_channels": 3,
40
+ "num_hidden_layers": 12,
41
+ "out_features": [
42
+ "stage12"
43
+ ],
44
+ "out_indices": [
45
+ 12
46
+ ],
47
+ "patch_size": 14,
48
+ "problem_type": "single_label_classification",
49
+ "qkv_bias": true,
50
+ "reshape_hidden_states": true,
51
+ "stage_names": [
52
+ "stem",
53
+ "stage1",
54
+ "stage2",
55
+ "stage3",
56
+ "stage4",
57
+ "stage5",
58
+ "stage6",
59
+ "stage7",
60
+ "stage8",
61
+ "stage9",
62
+ "stage10",
63
+ "stage11",
64
+ "stage12"
65
+ ],
66
+ "torch_dtype": "float32",
67
+ "transformers_version": "4.48.3",
68
+ "use_swiglu_ffn": false
69
+ }
logs/events.out.tfevents.1741874217.08c8ff51aa91.308.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e1baed6a1fd0392b89bb6630641982925bfebae85ec88a6dee42f15bc06a13f6
3
+ size 11372
logs/events.out.tfevents.1741879037.ba36c15cdafa.233.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8003dd6366b579180c209510c80db4e9b93589c7fca4de04a7f7b7bde21a31e3
3
+ size 17610
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:40f5e3a0d45c7a98b09fd408446f895797436a1b6fffffcf0a8ebea16f5091f9
3
+ size 346396816
preprocessor_config.json ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "crop_size": {
3
+ "height": 224,
4
+ "width": 224
5
+ },
6
+ "do_center_crop": true,
7
+ "do_convert_rgb": true,
8
+ "do_normalize": true,
9
+ "do_rescale": true,
10
+ "do_resize": true,
11
+ "image_mean": [
12
+ 0.485,
13
+ 0.456,
14
+ 0.406
15
+ ],
16
+ "image_processor_type": "BitImageProcessor",
17
+ "image_std": [
18
+ 0.229,
19
+ 0.224,
20
+ 0.225
21
+ ],
22
+ "resample": 3,
23
+ "rescale_factor": 0.00392156862745098,
24
+ "size": {
25
+ "shortest_edge": 256
26
+ }
27
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8a81f9c88a589168e7e400466762c90a6da127f45031727b631ccd3e5ab9c463
3
+ size 5496
training_metrics.xlsx ADDED
Binary file (8.33 kB). View file