ConvNextV2 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,23 @@
+---
+{}
+---
+# WD 1.4 ConvNextV2 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.3710, F1 = 0.6862`
+
+## 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:4e58abc2a26a4c49319c9977dbcd4f7eb92ea255f2ef115b3079040662aa7697
+size 476609

model.onnx

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

saved_model.pb

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

variables/variables.data-00000-of-00001

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

variables/variables.index

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