From 1e3800cc16ab9c69ae40e11f4374d76ef0350aa0 Mon Sep 17 00:00:00 2001 From: Přemysl Eric Janouch Date: Sun, 21 Jan 2024 10:38:46 +0100 Subject: deeptagger: fix Caformer By using the smaller resolution, it starts noticing 2girls, otherwise the output appears similar. --- deeptagger/download.sh | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) (limited to 'deeptagger/download.sh') 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' -- cgit v1.2.3-70-g09d2