a
Adu @ sr d dl Z d dlmZ d dlZd dlmZ d dlmZm Z m
Z
m Z d dl m
Z
d dlmZmZ G dd dejZdS ) N)Image)
processingsharedimagesdevices) Processed)optsstatec @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
Scriptc C s dS )Nz
SD upscale )selfr r :/var/www/html/stable-diffusion-webui/scripts/sd_upscale.pytitle
s z Script.titlec C s |S )Nr )r
is_img2imgr r r
show s z Script.showc C s| t d}t jddddd| dd}t jd d
d d d
| dd}t jddd tjD tjd jd| dd}||||gS )NzWill upscale the image by the selected scale factor; use width and height sliders to set tile size
r z Tile overlap@ overlap)minimummaximumsteplabelvalueelem_idg ?g @g?z Scale Factorg @ scale_factorUpscalerc S s g | ]
}|j qS r )name.0xr r r
z Script.ui..indexupscaler_index)r choicesr typer )grHTMLSliderr Radior sd_upscalersr )r r infor r r$ r r r
ui s
.z Script.uic C s t |tr$dd tjD | }t| tj| }||jd< |j |jd< d }|j
}|j d } t
| tj} |j dkr|j| ||j}
n| }
t t j|
|j|j|d} |j} |j}
d|_d |_d |_g }| j D ]"\}}}|D ]}| |d
qqt t || }||
t!_"t#d t | d t | j d d
d
t | j dt!j" d g }t$|
D ].}|| }||_
g }t$|D ]v}| |_||| |d | |_ d|d || dt!j" t!_%t&|}|d u r|j'}|j
d |_
||j 7 }qd}| j D ]N\}}}|D ]<}|t |k r*|| nt()d|j|jf|d
< |d7 }qqt *| }| | tj+r`t j,||j-d||j.tj/||d q`t0||||}|S )Nc S s g | ]}|j qS r )r lowerr r r r
r! r" z Script.run..zSD upscale overlapzSD upscale upscalerr None)tile_wtile_hr T z%SD upscaling will process a total of z images tiled as r z per upscale in a total of z batches.zBatch z out of RGB )r, p)1
isinstancestrr r+ r# r. r fix_seedextra_generation_paramsr seed init_imagesr flattenr img2img_background_colorscalerupscale data_pathr torch_gc
split_gridwidthheight
batch_sizen_iterdo_not_save_griddo_not_save_samplestilesappendmathceillenr job_countprintrangejobprocess_imagesr, r new combine_grid samples_save
save_imageoutpath_samplespromptsamples_formatr ) r r6 _r r$ r upscaler initial_infor; init_imgimggridrF Z
upscale_countwork_y_hrowtiledata batch_count
result_imagesnZ
start_seed work_resultsi processed image_indexcombined_imager r r
run sh
<
.
$z
Script.runN)__name__
__module__ __qualname__r r r- rn r r r r
r
s r
)rL modules.scriptsscriptsgradior' PILr modulesr r r r Zmodules.processingr Zmodules.sharedr r r
r r r r
s