|
|
|
import matplotlib.pyplot as plt |
|
from PIL import Image |
|
|
|
plt.rcParams["figure.figsize"] = (10, 5) |
|
plt.rcParams['figure.facecolor'] = 'white' |
|
|
|
|
|
def render_figure(model_name, fn): |
|
image_types = ['bird', 'human', 'room', 'vermeer'] |
|
|
|
def plot_row(axs, control_fn_prefix, output_fn_prefix, name, show_control=False): |
|
for i, ax in enumerate(axs): |
|
if i == 0: |
|
if show_control: |
|
ax.set_title(f'Control') |
|
ax.imshow(Image.open(f'{control_fn_prefix}.png')) |
|
else: |
|
ax.set_title(f'Seed={i-1} ({name})') |
|
ax.imshow(Image.open(f'{output_fn_prefix}_{i-1}.png')) |
|
|
|
fig, axs = plt.subplots( |
|
2 * len(image_types), 5, layout="constrained", figsize=(10, 5 * len(image_types))) |
|
for ax in axs.flatten(): |
|
ax.set_aspect('equal', 'box') |
|
ax.axis('off') |
|
|
|
pair_axs = [list(pair) for pair in zip(axs[::2], axs[1::2])] |
|
for image_type, pair_ax in zip(image_types, pair_axs): |
|
plot_row(pair_ax[0], |
|
f'./control_images/converted/control_{image_type}_{model_name}', |
|
f'./output_images/diffusers/output_{image_type}_{model_name}', |
|
'Diffusers', show_control=True) |
|
plot_row(pair_ax[1], |
|
f'./control_images/converted/control_{image_type}_{model_name}', |
|
f'./output_images/ref/output_{image_type}_{model_name}', |
|
'ref impl.') |
|
|
|
fig.suptitle(f'Model: {model_name}', fontsize=16) |
|
|
|
fig.savefig(fn, dpi=144) |
|
|
|
|
|
if __name__ == '__main__': |
|
model_names = ["canny", "normal", "depth", |
|
"openpose", "hed", "scribble", "mlsd", "seg"] |
|
for model in model_names: |
|
fn = f"plots/figure_{model}.png" |
|
render_figure(model, fn) |
|
print(fn) |
|
|
