Raditsoic commited on
Commit
4683e0c
1 Parent(s): f0a144a

feat: Release version 1.3

Browse files
.gitattributes CHANGED
@@ -35,3 +35,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
  models/trashnet_v1.2/trashnet_v1.2.keras filter=lfs diff=lfs merge=lfs -text
37
  trashnet.keras filter=lfs diff=lfs merge=lfs -text
 
 
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
  models/trashnet_v1.2/trashnet_v1.2.keras filter=lfs diff=lfs merge=lfs -text
37
  trashnet.keras filter=lfs diff=lfs merge=lfs -text
38
+ models/trashnet_v1.3/trashnet_v1.3.keras filter=lfs diff=lfs merge=lfs -text
models/trashnet_v1.3/README.md ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ datasets:
3
+ - garythung/trashnet
4
+ metrics:
5
+ - accuracy
6
+ pipeline_tag: image-classification
7
+ ---
8
+
9
+ # Model Version 1.3
10
+ Created on: 2024-12-22 15:00:20
11
+
12
+ ```
13
+
14
+ ## Metrics
15
+ - *train_loss*: 0.7903
16
+ - *train_accuracy*: 0.7188
17
+ - *val_loss*: 0.6419
18
+ - *val_accuracy*: 0.7671
19
+
20
+ ## Configuration
21
+ - **architecture**: 3 Convolutional Layers, 2 Dense Layers, Global Average Pooling, Dropout 0.3
22
+ - **input_size**: (224, 224, 3)
23
+ - **num_classes**: 6
24
+ - **augmentation**: {'rotation_range': 20, 'width_shift_range': 0.2, 'height_shift_range': 0.2, 'horizontal_flip': True, 'vertical_flip': True, 'shear_range': 0.2, 'brightness_range': [0.8, 1.2]}
25
+ - **optimizer**: adam
26
+ - **learning_rate**: 0.001
27
+ - **batch_size**: 32
28
+ - **epochs**: 50
models/trashnet_v1.3/config.json ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architecture": "3 Convolutional Layers, 2 Dense Layers, Global Average Pooling, Dropout 0.3",
3
+ "input_size": [
4
+ 224,
5
+ 224,
6
+ 3
7
+ ],
8
+ "num_classes": 6,
9
+ "augmentation": {
10
+ "rotation_range": 20,
11
+ "width_shift_range": 0.2,
12
+ "height_shift_range": 0.2,
13
+ "horizontal_flip": true,
14
+ "vertical_flip": true,
15
+ "shear_range": 0.2,
16
+ "brightness_range": [
17
+ 0.8,
18
+ 1.2
19
+ ]
20
+ },
21
+ "optimizer": "adam",
22
+ "learning_rate": 0.001,
23
+ "batch_size": 32,
24
+ "epochs": 50
25
+ }
models/trashnet_v1.3/metrics.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "train_loss": 0.7903071641921997,
3
+ "train_accuracy": 0.71875,
4
+ "val_loss": 0.6418748497962952,
5
+ "val_accuracy": 0.7671371102333069
6
+ }
models/trashnet_v1.3/trashnet_v1.3.keras ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9fc22638d0bdff852b919e5825ac284a6d67dfb3eba287d67f154d717043f028
3
+ size 1408450