ConvNext Tagger v2.0

.gitattributes

@@ -32,3 +32,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+variables/variables.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text
+variables/variables.index filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,20 @@
+# WD 1.4 ConvNext Tagger V2
+
+Supports ratings, characters and general tags.
+
+Trained using https://github.com/SmilingWolf/SW-CV-ModelZoo.  
+TPUs used for training kindly provided by the [TRC program](https://sites.research.google/trc/about/).
+
+## Dataset
+Last image id: 5944504
+Trained on Danbooru images with IDs modulo 0000-0899.  
+Validated on images with IDs modulo 0950-0999.  
+Images with less than 10 general tags were filtered out.  
+Tags with less than 600 images were filtered out.
+
+## Validation results
+`P=R: threshold = 0.3685, F1: 0.6810`
+
+## Final words
+Subject to change and updates.  
+Downstream users are encouraged to use tagged releases rather than relying on the head of the repo.

keras_metadata.pb

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

model.onnx

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

saved_model.pb

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

variables/variables.data-00000-of-00001

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

variables/variables.index

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