diff options
| author | Přemysl Eric Janouch <p@janouch.name> | 2024-01-19 20:00:05 +0100 |
|---|---|---|
| committer | Přemysl Eric Janouch <p@janouch.name> | 2024-01-19 20:02:24 +0100 |
| commit | 4131bc5d3130e876d021e1cfa7fae4d3c3b86626 (patch) | |
| tree | fd3fbf3becf1cf53f4836ed5dc1296ad4af3fd05 /deeptagger | |
| parent | fd5e3bb166638a9666ce632d5965f46745d320a9 (diff) | |
| download | gallery-4131bc5d3130e876d021e1cfa7fae4d3c3b86626.tar.gz gallery-4131bc5d3130e876d021e1cfa7fae4d3c3b86626.tar.xz gallery-4131bc5d3130e876d021e1cfa7fae4d3c3b86626.zip | |
Add benchmarks against WDMassTagger
Diffstat (limited to 'deeptagger')
| -rw-r--r-- | deeptagger/README.adoc | 20 |
1 files changed, 18 insertions, 2 deletions
diff --git a/deeptagger/README.adoc b/deeptagger/README.adoc index 9a4d3aa..7b338af 100644 --- a/deeptagger/README.adoc +++ b/deeptagger/README.adoc @@ -142,7 +142,7 @@ The kernel panic was repeatable. GPU inference ~~~~~~~~~~~~~ -[cols="<,>,>", options=header] +[cols="<2,>1,>1", options=header] |=== |Model|Batch size|Time |DeepDanbooru|1|24 s @@ -177,7 +177,7 @@ GPU inference CPU inference ~~~~~~~~~~~~~ -[cols="<,>,>", options=header] +[cols="<2,>1,>1", options=header] |=== |Model|Batch size|Time |DeepDanbooru|8|54 s @@ -210,3 +210,19 @@ CPU inference |ML-Danbooru Caformer dec-5-97527|8|241 s |ML-Danbooru Caformer dec-5-97527|1|262 s |=== + +Comparison with WDMassTagger +---------------------------- +Using CUDA, on the same Linux computer as above, on a sample of 6352 images. +We're a bit slower, depending on the model. +Batch sizes of 16 and 32 give practically equivalent results for both. + +[cols="<,>,>,>", options="header,autowidth"] +|=== +|Model|WDMassTagger|deeptagger (batch)|Ratio +|wd-v1-4-convnext-tagger-v2 |1:18 |1:55 |68 % +|wd-v1-4-convnextv2-tagger-v2 |1:20 |2:10 |62 % +|wd-v1-4-moat-tagger-v2 |1:22 |1:52 |73 % +|wd-v1-4-swinv2-tagger-v2 |1:28 |1:34 |94 % +|wd-v1-4-vit-tagger-v2 |1:16 |1:22 |93 % +|=== |