Spaces:
Running
Running
File size: 2,357 Bytes
dc4014d ccea513 dc4014d ccea513 dc4014d ccea513 dc4014d ccea513 dc4014d ccea513 0c9e779 ccea513 dc4014d ccea513 dc4014d ccea513 dc4014d ccea513 dc4014d ccea513 0c9e779 |
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 56 57 58 59 60 61 62 63 64 |
import gradio as gr
import torch
from utils import *
import cv2
import numpy as np
import matplotlib.pyplot as plt
import io
torch.hub.download_url_to_file(
'https://github.com/aalto-ui/aim/raw/aim2/backend/data/tests/input_values/wikipedia.org_website.png',
'wikipedia.org_website.png')
torch.hub.download_url_to_file(
'https://github.com/aalto-ui/aim/raw/aim2/backend/data/tests/input_values/aalto.fi_website.png',
'aalto.fi_website.png')
def inference(img, template, angel):
color_image = cv2.imread(img.name, cv2.IMREAD_COLOR)
HSV_image = cv2.cvtColor(color_image, cv2.COLOR_BGR2HSV)
selected_harmomic_scheme = HarmonicScheme(str(template), int(angel))
new_HSV_image = selected_harmomic_scheme.hue_shifted(HSV_image, num_superpixels=-1)
# Compute shifted histogram
histo_1 = count_hue_histogram(HSV_image)
histo_2 = count_hue_histogram(new_HSV_image)
# Create Hue Plots
fig1 = plothis(histo_1, selected_harmomic_scheme, "Source Hue")
fig_1_cv = get_img_from_fig(fig1)
fig2 = plothis(histo_2, selected_harmomic_scheme, "Target Hue")
fig_2_cv = get_img_from_fig(fig2)
# Convert HSV to BGR
result_image = cv2.cvtColor(new_HSV_image, cv2.COLOR_HSV2BGR)
cv2.imwrite('fig_1_cv.jpg', fig_1_cv)
cv2.imwrite('fig_2_cv.jpg', fig_2_cv)
cv2.imwrite('result_image.jpg', result_image)
return ['result_image.jpg', 'fig_1_cv.jpg', 'fig_2_cv.jpg']
title = 'Color Harmonization'
description = 'Compute Color Harmonization with Different Templates'
article = "<p style='text-align: center'></p>"
examples = [['wikipedia.org_website.png'], ['aalto.fi_website.png']]
# css = ".output_image, .input_image {height: 40rem !important; width: 100% !important;}"
gr.Interface(
inference,
[gr.inputs.Image(type='file', label='Input'),
gr.inputs.Dropdown(["X", "Y", "T", "I", "mirror_L", "L", "V", "i"],
default="X",
label="Template"),
gr.inputs.Slider(0, 359, label="Angle")],
[gr.outputs.Image(type='file', label='Color Harmonization of Output Image'),
gr.outputs.Image(type='file', label='Source Hue'),
gr.outputs.Image(type='file', label='Target Hue'),],
title=title,
description=description,
article=article,
examples=examples,
# css=css,
).launch(debug=True, enable_queue=True) |