diff options
| author | Přemysl Eric Janouch <p@janouch.name> | 2024-01-21 10:38:46 +0100 |
|---|---|---|
| committer | Přemysl Eric Janouch <p@janouch.name> | 2024-01-21 10:38:46 +0100 |
| commit | 1e3800cc16ab9c69ae40e11f4374d76ef0350aa0 (patch) | |
| tree | 6f195f141d95af06869b30c9522c3e552e8a6fc4 /deeptagger/download.sh | |
| parent | d37e9e821a63f5e0dd8dfd878df782c5c745f0e8 (diff) | |
| download | gallery-1e3800cc16ab9c69ae40e11f4374d76ef0350aa0.tar.gz gallery-1e3800cc16ab9c69ae40e11f4374d76ef0350aa0.tar.xz gallery-1e3800cc16ab9c69ae40e11f4374d76ef0350aa0.zip | |
deeptagger: fix Caformer
By using the smaller resolution, it starts noticing 2girls,
otherwise the output appears similar.
Diffstat (limited to 'deeptagger/download.sh')
| -rwxr-xr-x | deeptagger/download.sh | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/deeptagger/download.sh b/deeptagger/download.sh index 29f651e..7336f35 100755 --- a/deeptagger/download.sh +++ b/deeptagger/download.sh @@ -115,7 +115,7 @@ wd14() { # These models are an undocumented mess, thus using ONNX preconversions. mldanbooru() { - local name=$1 basename=$2 + local name=$1 size=$2 basename=$3 status "$name" if ! [ -d ml-danbooru-onnx ] @@ -138,7 +138,7 @@ mldanbooru() { channels=rgb normalize=true pad=stretch - size=640 + size=$size interpret=sigmoid END } @@ -157,5 +157,7 @@ wd14 'WD v1.4 SwinV2 v2' 'SmilingWolf/wd-v1-4-swinv2-tagger-v2' wd14 'WD v1.4 MOAT v2' 'SmilingWolf/wd-v1-4-moat-tagger-v2' # As suggested by author https://github.com/IrisRainbowNeko/ML-Danbooru-webui -mldanbooru 'ML-Danbooru Caformer dec-5-97527' 'ml_caformer_m36_dec-5-97527.onnx' -mldanbooru 'ML-Danbooru TResNet-D 6-30000' 'TResnet-D-FLq_ema_6-30000.onnx' +mldanbooru 'ML-Danbooru Caformer dec-5-97527' \ + 448 'ml_caformer_m36_dec-5-97527.onnx' +mldanbooru 'ML-Danbooru TResNet-D 6-30000' \ + 640 'TResnet-D-FLq_ema_6-30000.onnx' |