laplace.py

import cv2
import pylab
import numpy as np
import skimage.filters.edges
import matplotlib.pyplot as plt


def laplace(image, path='', save=False, show=False):
    '''
    Deteccion de bordes mediante el operador de Laplace.
    '''
    # convertir a imagen en escala de grises
    image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

    # detectar los bordes mediante el metodo Laplace
    laplace = cv2.Laplacian(image, cv2.CV_64F)
    laplace = np.uint8(np.absolute(laplace))
    # laplace1 = skimage.filters.edges.laplace(image)

    if show:
        show_image(laplace)
        # show_image(laplace1)
        show_images([image, laplace])
    if save:
        name = get_name_image(path)
        save_image(laplace, f'../../img/{name}_laplace.png')

    return laplace


def get_name_image(path):
    '''
    Devuelve el nombre de la imagen segun la ruta.
    '''
    name_extention = path.split('/')[-1]
    name, _ = name_extention.split('.')
    return name


def load_image(path):
    '''
    Carga la imagen desde la ruta especificada.
    '''
    img = cv2.imread(path)
    return img


def show_image(image):
    '''
    Muestra la imagen especificada.
    '''
    plt.imshow(image, cmap='gray')
    plt.show()


def show_images(images):
    '''
    Muestra las imagenes especificadas.
    '''
    count_image = len(images)
    _, axs = plt.subplots(1, count_image)

    for idx, image in enumerate(images):
        axs[idx].imshow(image, cmap='gray')

    plt.show()


def save_image(image, path):
    '''
    Guarda la imagen en la ruta especificada.
    '''
    pylab.gray()
    pylab.imsave(path, image)


def main():

    path = '../../img/lena.tif'
    image = load_image(path)

    laplace(image, path=path, save=True, show=True)


if __name__ == '__main__':
    main()