feat: Update main with the latest model

.gitattributes

@@ -34,3 +34,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 models/trashnet_v1.2/trashnet_v1.2.keras filter=lfs diff=lfs merge=lfs -text
+trashnet.keras filter=lfs diff=lfs merge=lfs -text

README.md

@@ -1,16 +1,28 @@
----
-datasets:
-- garythuang/trashnet
-metrics:
-- accuracy
-pipeline_tag: image-classification
----
-
-# Model Version 1.1
-Created on: 2024-12-20 19:53:42
-
-## Metrics
-- *train_loss*: 0.7815648026353731
-- *train_accuracy*: 72.17412812268118
-- *val_loss*: 0.9300829283893108
-- *val_accuracy*: 67.06231454005935
+---
+datasets:
+- garythung/trashnet
+metrics:
+- accuracy
+pipeline_tag: image-classification
+---
+
+# Model Version 1.2
+Created on: 2024-12-21 21:58:40
+
+```
+
+## Metrics
+- *train_loss*: 0.5705
+- *train_accuracy*: 0.7812
+- *val_loss*: 0.5975
+- *val_accuracy*: 0.7752
+
+## Configuration
+- **architecture**: Tensorflow CNN
+- **input_size**: (224, 224, 3)
+- **num_classes**: 6
+- **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]}
+- **optimizer**: adam
+- **learning_rate**: 0.001
+- **batch_size**: 32
+- **epochs**: 50

config.json

@@ -1,20 +1,25 @@
-{
-    "architecture": "Custom CNN",
-    "input_size": [
-        224,
-        224
-    ],
-    "num_classes": 6,
-    "transformations": [
-        "RandomHorizontalFlip",
-        "RandomRotation(15)",
-        "ColorJitter(brightness=0.2, contrast=0.2)",
-        "Resize(224x224)",
-        "ToTensor",
-        "Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])"
-    ],
-    "optimizer": "Adam",
-    "learning_rate": 0.001,
-    "batch_size": 32,
-    "epochs": 10
+{
+    "architecture": "Tensorflow CNN",
+    "input_size": [
+        224,
+        224,
+        3
+    ],
+    "num_classes": 6,
+    "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
+        ]
+    },
+    "optimizer": "adam",
+    "learning_rate": 0.001,
+    "batch_size": 32,
+    "epochs": 50
 }

metrics.json

@@ -1,6 +1,6 @@
-{
-    "train_loss": 0.7815648026353731,
-    "train_accuracy": 72.17412812268118,
-    "val_loss": 0.9300829283893108,
-    "val_accuracy": 67.06231454005935
+{
+    "train_loss": 0.5704573392868042,
+    "train_accuracy": 0.78125,
+    "val_loss": 0.5974659323692322,
+    "val_accuracy": 0.7752016186714172
 }

trashnet.keras

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e51e07f0c49eb33e234473e74cfa0175a0e773d4376b0a2dc43d1c2090dda072
+size 155352521