diff options
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' |