Hugging Face's logo

PhoenixAscencio
/
HWtagger

TF-Keras
ONNX
Model card Files Community
PhoenixAscencio commited on
Commit
07f4641
·
verified ·
1 Parent(s): 6d53604

Upload 20 files

Browse files
Files changed (21) hide show
  1. .gitattributes +2 -0
  2. models/ML-danbooru-Caformer/class.json +0 -0
  3. models/ML-danbooru-Caformer/model.ckpt +3 -0
  4. models/SwinV2/keras_metadata.pb +3 -0
  5. models/SwinV2/saved_model.pb +3 -0
  6. models/SwinV2/selected_tags.csv +0 -0
  7. models/SwinV2/tmpugou97nr +3 -0
  8. models/SwinV2/variables/variables.data-00000-of-00001 +3 -0
  9. models/SwinV2/variables/variables.index +0 -0
  10. models/SwinV2v3/config.json +37 -0
  11. models/SwinV2v3/model.onnx +3 -0
  12. models/SwinV2v3/selected_tags.csv +0 -0
  13. models/SwinV2v3/sw_jax_cv_config.json +19 -0
  14. models/anime_aesthetic/model.onnx +3 -0
  15. models/anime_aesthetic/samples_x.npy +3 -0
  16. models/anime_aesthetic/samples_y.npy +3 -0
  17. models/anime_aesthetic/swinv2pv3_v0_448_ls0.2_x_meta.json +15 -0
  18. models/anime_aesthetic/swinv2pv3_v0_448_ls0.2_x_metrics.json +7 -0
  19. models/anime_classifier/mobilenetv3_v1.3_dist_meta.json +11 -0
  20. models/anime_classifier/mobilenetv3_v1.3_dist_metrics.json +7 -0
  21. models/anime_classifier/model.onnx +3 -0
.gitattributes CHANGED
@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ models/SwinV2/tmpugou97nr filter=lfs diff=lfs merge=lfs -text
37
+ models/SwinV2/variables/variables.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text
models/ML-danbooru-Caformer/class.json ADDED
The diff for this file is too large to render. See raw diff
 
models/ML-danbooru-Caformer/model.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:587dc60e0df43e5802b465d39f7ae57b8827c76040eb30d33d404b28c9ade3b2
3
+ size 602119197
models/SwinV2/keras_metadata.pb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f48bd1aa100ea3db529c42658dc2b5505befef0182035d7d7617e1b0ee7b9feb
3
+ size 447720
models/SwinV2/saved_model.pb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dae9e67ee047727873c7356c1fad95232819abc3e9714b20f878ebc3175ed36c
3
+ size 37577107
models/SwinV2/selected_tags.csv ADDED
The diff for this file is too large to render. See raw diff
 
models/SwinV2/tmpugou97nr ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5624a48d5893fb398db14f3d66fc4436f08559ff9ff60b7a88e731c4bc15988c
3
+ size 31457280
models/SwinV2/variables/variables.data-00000-of-00001 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b56699435b5f9754ed86b0a0d087c3670097fa2cbcdfce89a0f74731fbbb227a
3
+ size 385176929
models/SwinV2/variables/variables.index ADDED
Binary file (24.2 kB). View file
 
models/SwinV2v3/config.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architecture": "swinv2_base_window8_256",
3
+ "num_classes": 10861,
4
+ "num_features": 1024,
5
+ "global_pool": "avg",
6
+ "model_args": {
7
+ "act_layer": "gelu_tanh",
8
+ "img_size": 448,
9
+ "window_size": 14
10
+ },
11
+ "pretrained_cfg": {
12
+ "custom_load": false,
13
+ "input_size": [
14
+ 3,
15
+ 448,
16
+ 448
17
+ ],
18
+ "fixed_input_size": false,
19
+ "interpolation": "bicubic",
20
+ "crop_pct": 1.0,
21
+ "crop_mode": "center",
22
+ "mean": [
23
+ 0.5,
24
+ 0.5,
25
+ 0.5
26
+ ],
27
+ "std": [
28
+ 0.5,
29
+ 0.5,
30
+ 0.5
31
+ ],
32
+ "num_classes": 10861,
33
+ "pool_size": null,
34
+ "first_conv": null,
35
+ "classifier": null
36
+ }
37
+ }
models/SwinV2v3/model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e6774bff34d43bd49f75a47db4ef217dce701c9847b546523eb85ff6dbba1db1
3
+ size 467460978
models/SwinV2v3/selected_tags.csv ADDED
The diff for this file is too large to render. See raw diff
 
models/SwinV2v3/sw_jax_cv_config.json ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "image_size": 448,
3
+ "model_name": "swinv2_base",
4
+ "model_args": {
5
+ "image_size": 448,
6
+ "patch_size": 4,
7
+ "in_chans": 3,
8
+ "num_classes": 10861,
9
+ "embed_dim": 128,
10
+ "window_size": 14,
11
+ "mlp_ratio": 4.0,
12
+ "qkv_bias": true,
13
+ "drop_rate": 0.0,
14
+ "attn_drop_rate": 0.0,
15
+ "drop_path_rate": 0.1,
16
+ "patch_norm": true,
17
+ "layer_norm_eps": 1e-05
18
+ }
19
+ }
models/anime_aesthetic/model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:174a988ff458e84bd28c2a71cd22a5e7e160ec2518e501efbdfc3de1d09d82a2
3
+ size 418260629
models/anime_aesthetic/samples_x.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d01afa2e7ddcc2e7ee7024576d68ee492a7304cbbcca9ec9a3a420000e042007
3
+ size 102472
models/anime_aesthetic/samples_y.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b836ef1ea9424a2d325c6d71a4f00212f8cd565a9b9f9df7b178ab32e87a81d8
3
+ size 102472
models/anime_aesthetic/swinv2pv3_v0_448_ls0.2_x_meta.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "drop_path_rate": 0.4,
3
+ "img_size": 448,
4
+ "labels": [
5
+ "masterpiece",
6
+ "best",
7
+ "great",
8
+ "good",
9
+ "normal",
10
+ "low",
11
+ "worst"
12
+ ],
13
+ "name": "hf-hub:SmilingWolf/wd-swinv2-tagger-v3",
14
+ "pretrained": true
15
+ }
models/anime_aesthetic/swinv2pv3_v0_448_ls0.2_x_metrics.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "AUC": 0.8213613069365249,
3
+ "accuracy": 0.40876370669619877,
4
+ "loss": 1.6459708088109346,
5
+ "mAP": 0.4248220760347013,
6
+ "step": 18
7
+ }
models/anime_classifier/mobilenetv3_v1.3_dist_meta.json ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "labels": [
3
+ "3d",
4
+ "bangumi",
5
+ "comic",
6
+ "illustration",
7
+ "not_painting"
8
+ ],
9
+ "name": "mobilenetv3_large_100",
10
+ "pretrained": true
11
+ }
models/anime_classifier/mobilenetv3_v1.3_dist_metrics.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "AUC": 0.9973236276399039,
3
+ "accuracy": 0.9641284641284641,
4
+ "cls_loss": 0.06461880440611652,
5
+ "mAP": 0.9910443784192255,
6
+ "step": 50
7
+ }
models/anime_classifier/model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fa233c04467cba4c68ecabd0065ad169c6fe867dc19799627cf5bd566a97d115
3
+ size 16837808