mamorecab's picture
Update app.py
66479e0
import gradio as gr
import cv2
from fastai.vision.all import *
import imutils
# Definimos una función que se encarga de llevar a cabo las predicciones
def panoramica(img1,img2):
lista = []
pilImg1=PIL.Image.open(img1).convert('RGB')
open_cv_image = numpy.array(pilImg1)
lista.append(open_cv_image)
pilImg2=PIL.Image.open(img2).convert('RGB')
open_cv_image2 = numpy.array(pilImg2)
lista.append(open_cv_image2 )
# for i in imagenes:
# lista.append(PILImage.create(i))
stitcher = cv2.createStitcher() if imutils.is_cv3() else cv2.Stitcher_create()
(estado, pan) = stitcher.stitch(lista)
return pan
# Creamos la interfaz y la lanzamos.
gr.Interface(fn=panoramica, inputs=[gr.inputs.Image(),gr.inputs.Image()], outputs=gr.outputs.Image(),examples=['edificio1.jpg','edificio2.jpg']).launch(share=False)