diff options
| author | DepFA <35278260+dfaker@users.noreply.github.com> | 2022-10-09 19:41:22 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-10-09 19:41:22 +0000 |
| commit | fa0c5eb81b72bc1e562d0b9bbd92f30945d78b4e (patch) | |
| tree | e80152606a00730bf4dd11167f4ebdd2d70f50ca /modules/images.py | |
| parent | cd8673bd9b2e59bddefee8d307340d643695fe11 (diff) | |
| download | stable-diffusion-webui-gfx803-fa0c5eb81b72bc1e562d0b9bbd92f30945d78b4e.tar.gz stable-diffusion-webui-gfx803-fa0c5eb81b72bc1e562d0b9bbd92f30945d78b4e.tar.bz2 stable-diffusion-webui-gfx803-fa0c5eb81b72bc1e562d0b9bbd92f30945d78b4e.zip | |
Add pretty image captioning functions
Diffstat (limited to 'modules/images.py')
| -rw-r--r-- | modules/images.py | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/modules/images.py b/modules/images.py index 29c5ee24..10963dc7 100644 --- a/modules/images.py +++ b/modules/images.py @@ -428,3 +428,34 @@ def save_image(image, path, basename, seed=None, prompt=None, extension='png', i file.write(info + "\n")
return fullfn
+
+def addCaptionLines(lines,image,initialx,textfont):
+ draw = ImageDraw.Draw(image)
+ hstart =initialx
+ for fill,line in lines:
+ fontSize = 32
+ font = ImageFont.truetype(textfont, fontSize)
+ _,_,w, h = draw.textbbox((0,0),line,font=font)
+ fontSize = min( int(fontSize * ((image.size[0]-35)/w) ), 28)
+ font = ImageFont.truetype(textfont, fontSize)
+ _,_,w,h = draw.textbbox((0,0),line,font=font)
+ draw.text(((image.size[0]-w)/2,hstart), line, font=font, fill=fill)
+ hstart += h
+ return hstart
+
+def captionImge(image,prelines,postlines,background=(51, 51, 51),font=None):
+ if font is None:
+ try:
+ font = ImageFont.truetype(opts.font or Roboto, fontsize)
+ font = opts.font or Roboto
+ except Exception:
+ font = Roboto
+
+ sampleImage = image
+ background = Image.new("RGBA", (sampleImage.size[0],sampleImage.size[1]+1024), background)
+ hoffset = addCaptionLines(prelines,background,5,font)+16
+ background.paste(sampleImage,(0,hoffset))
+ hoffset = hoffset+sampleImage.size[1]+8
+ hoffset = addCaptionLines(postlines,background,hoffset,font)
+ background = background.crop((0,0,sampleImage.size[0],hoffset+8))
+ return background
|