File size: 1,723 Bytes
db57927 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
import modules.scripts as scripts
import gradio as gr
import os
from modules import images
from modules.processing import process_images, Processed
from modules.processing import Processed
from modules.shared import opts, cmd_opts, state
class Script(scripts.Script):
def title(self):
return "Unsharp Mask"
#only show in img2img tab
def show(self, is_img2img):
return is_img2img
#Gradio interface parameters
def ui(self, is_img2img):
save = gr.Checkbox(False, label="Save original and effect")
umradius = gr.Slider(minimum=0.0, maximum=1000.0, step=1, value=0, label="Radius")
umpercent = gr.Slider(minimum=0.0, maximum=500.0, step=1, value=0, label="Percent")
umthreshold = gr.Slider(minimum=0.0, maximum=255.0, step=1, value=0, label="Threshold")
return [save, umradius, umpercent, umthreshold]
#processed object
def run(self, p, save, umradius, umpercent, umthreshold):
#perform unsharp mask operation
def unsharp_mask(im, umradius, umpercent, umthreshold):
from PIL import Image, ImageFilter
raf = im
raf = raf.filter(filter=ImageFilter.UnsharpMask(radius = umradius, percent = umpercent, threshold = umthreshold))
return raf
# If save is off, save with prefix filename
basename = ""
if(save):
basename += "unsharpmask_"
else:
p.do_not_save_samples = True
#process images
proc = process_images(p)
for i in range(len(proc.images)):
proc.images[i] = unsharp_mask(proc.images[i], umradius, umpercent, umthreshold)
images.save_image(proc.images[i], p.outpath_samples, basename,
proc.seed + i, proc.prompt, opts.samples_format, info= proc.info, p=p)
return proc
|