Migrasi batch otomatis

README.md

@@ -0,0 +1,27 @@
+---
+license: mit
+datasets:
+- deepghs/monochrome_danbooru
+pipeline_tag: image-classification
+library_name: dghs-imgutils
+tags:
+- art
+- onnx
+- safetensors
+- emb_attachments
+- wdtaggers
+---
+
+This is an experiment on DeepGHS's new tagger attachment training framework [deepghs/emb_attachments](https://github.com/deepghs/emb_attachments).
+
+Just some technical validation, not recommended for use in production.
+
+(2025.3.7: This feature is still WIP, you can take a look at [this PR](https://github.com/deepghs/imgutils/pull/152) and [this documentation](https://dghs-imgutils.deepghs.org/dev/attachments/api_doc/tagging/wd14.html#get-wd14-tags), you will like it.)
+
+|            Name             |                 Encoder                  |    Problem     |  FLOPS  |  Params  |  AUC   |  accuracy  |  loss   |  mAP   |       Labels       |
+|:---------------------------:|:----------------------------------------:|:--------------:|:-------:|:--------:|:------:|:----------:|:-------:|:------:|:------------------:|
+| mlp_layer1_seed114514_dist8 | wdtagger:SmilingWolf/wd-swinv2-tagger-v3 | classification | 525.3K  |  525.8K  | 0.9937 |   0.9743   | 0.04483 | 0.9937 | monochrome, normal |
+|   mlp_layer1_seedx_dist8    | wdtagger:SmilingWolf/wd-swinv2-tagger-v3 | classification | 525.3K  |  525.8K  | 0.9946 |   0.9732   | 0.04058 | 0.9947 | monochrome, normal |
+|   mlp_layer1_seed3_dist8    | wdtagger:SmilingWolf/wd-swinv2-tagger-v3 | classification | 525.3K  |  525.8K  | 0.9956 |   0.9699   | 0.05301 | 0.9957 | monochrome, normal |
+|      mlp_layer1_seed1       | wdtagger:SmilingWolf/wd-swinv2-tagger-v3 | classification | 525.3K  |  525.8K  | 0.9948 |   0.9652   | 0.06904 | 0.995  | monochrome, normal |
+

mlp_layer1_seed1/events.out.tfevents.1741272460.77cc9058c7a7b07cddf3cf65aa1202cb5e26bda5001d182067e0d90f.661707.0

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

mlp_layer1_seed1/meta.json

@@ -0,0 +1,30 @@
+{
+    "encoder_model": "wdtagger:SmilingWolf/wd-swinv2-tagger-v3",
+    "flops": 525312.0,
+    "init_params": {
+        "in_dims": 1024,
+        "layers": [
+            0.5
+        ],
+        "out_dims": 2
+    },
+    "model_type": "mlp",
+    "params": 525826.0,
+    "problem": {
+        "labels": [
+            "monochrome",
+            "normal"
+        ],
+        "type": "classification"
+    },
+    "train": {
+        "batch_size": 16,
+        "key_metric": "AUC",
+        "learning_rate": 0.001,
+        "loss": "focal",
+        "max_epochs": 100,
+        "processes": 1,
+        "seed": 1,
+        "weight_decay": 0.001
+    }
+}

mlp_layer1_seed1/metrics.json

@@ -0,0 +1,6 @@
+{
+    "AUC": 0.9948011725015209,
+    "accuracy": 0.9652466367713004,
+    "loss": 0.06903827775926219,
+    "mAP": 0.9949569298089084
+}

mlp_layer1_seed1/model.onnx

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

mlp_layer1_seed1/model.safetensors

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

mlp_layer1_seed114514_dist8/events.out.tfevents.1742473121.197caa7ba88141226195c99eeafb8490b47d74629daea96443dcc28f.263082.0

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

mlp_layer1_seed114514_dist8/meta.json

@@ -0,0 +1,30 @@
+{
+    "encoder_model": "wdtagger:SmilingWolf/wd-swinv2-tagger-v3",
+    "flops": 525312.0,
+    "init_params": {
+        "in_dims": 1024,
+        "layers": [
+            0.5
+        ],
+        "out_dims": 2
+    },
+    "model_type": "mlp",
+    "params": 525826.0,
+    "problem": {
+        "labels": [
+            "monochrome",
+            "normal"
+        ],
+        "type": "classification"
+    },
+    "train": {
+        "batch_size": 16,
+        "key_metric": "AUC",
+        "learning_rate": 0.001,
+        "loss": "focal",
+        "max_epochs": 100,
+        "processes": 8,
+        "seed": 114514,
+        "weight_decay": 0.001
+    }
+}

mlp_layer1_seed114514_dist8/metrics.json

@@ -0,0 +1,6 @@
+{
+    "AUC": 0.9937066491270031,
+    "accuracy": 0.9743303571428571,
+    "loss": 0.04483043490576425,
+    "mAP": 0.9937499079664445
+}

mlp_layer1_seed114514_dist8/model.onnx

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

mlp_layer1_seed114514_dist8/model.safetensors

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

mlp_layer1_seed3_dist8/events.out.tfevents.1742459370.197caa7ba88141226195c99eeafb8490b47d74629daea96443dcc28f.35842.0

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

mlp_layer1_seed3_dist8/meta.json

@@ -0,0 +1,30 @@
+{
+    "encoder_model": "wdtagger:SmilingWolf/wd-swinv2-tagger-v3",
+    "flops": 525312.0,
+    "init_params": {
+        "in_dims": 1024,
+        "layers": [
+            0.5
+        ],
+        "out_dims": 2
+    },
+    "model_type": "mlp",
+    "params": 525826.0,
+    "problem": {
+        "labels": [
+            "monochrome",
+            "normal"
+        ],
+        "type": "classification"
+    },
+    "train": {
+        "batch_size": 16,
+        "key_metric": "AUC",
+        "learning_rate": 0.001,
+        "loss": "focal",
+        "max_epochs": 100,
+        "processes": 8,
+        "seed": 3,
+        "weight_decay": 0.001
+    }
+}

mlp_layer1_seed3_dist8/metrics.json

@@ -0,0 +1,6 @@
+{
+    "AUC": 0.9956101012516942,
+    "accuracy": 0.9698660714285714,
+    "loss": 0.05300576690310014,
+    "mAP": 0.9957305722508567
+}

mlp_layer1_seed3_dist8/model.onnx

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

mlp_layer1_seed3_dist8/model.safetensors

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

mlp_layer1_seedx_dist8/events.out.tfevents.1742461859.197caa7ba88141226195c99eeafb8490b47d74629daea96443dcc28f.120820.0

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

mlp_layer1_seedx_dist8/meta.json

@@ -0,0 +1,30 @@
+{
+    "encoder_model": "wdtagger:SmilingWolf/wd-swinv2-tagger-v3",
+    "flops": 525312.0,
+    "init_params": {
+        "in_dims": 1024,
+        "layers": [
+            0.5
+        ],
+        "out_dims": 2
+    },
+    "model_type": "mlp",
+    "params": 525826.0,
+    "problem": {
+        "labels": [
+            "monochrome",
+            "normal"
+        ],
+        "type": "classification"
+    },
+    "train": {
+        "batch_size": 16,
+        "key_metric": "AUC",
+        "learning_rate": 0.001,
+        "loss": "focal",
+        "max_epochs": 100,
+        "processes": 8,
+        "seed": 353193873,
+        "weight_decay": 0.001
+    }
+}

mlp_layer1_seedx_dist8/metrics.json

@@ -0,0 +1,6 @@
+{
+    "AUC": 0.9946334608945229,
+    "accuracy": 0.9732142857142857,
+    "loss": 0.04057957721254622,
+    "mAP": 0.9947467799487766
+}

mlp_layer1_seedx_dist8/model.onnx

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

mlp_layer1_seedx_dist8/model.safetensors

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