SwinV2 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 SwinV2 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.3771, F1 = 0.6854`
+
+## 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:f48bd1aa100ea3db529c42658dc2b5505befef0182035d7d7617e1b0ee7b9feb
+size 447720

model.onnx

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

saved_model.pb

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

variables/variables.data-00000-of-00001

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

variables/variables.index

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