feat: Release version 1.1

models/trashnet_v1.1/README.md

@@ -0,0 +1,16 @@
+---
+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

models/trashnet_v1.1/config.json

@@ -0,0 +1,20 @@
+{
+    "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
+}

models/trashnet_v1.1/metrics.json

@@ -0,0 +1,6 @@
+{
+    "train_loss": 0.7815648026353731,
+    "train_accuracy": 72.17412812268118,
+    "val_loss": 0.9300829283893108,
+    "val_accuracy": 67.06231454005935
+}

models/trashnet_v1.1/trashnet_v1.1.pth

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