import numpy as np
import cv2 as cv
import pandas as pd
import os
from glob import glob
from matplotlib import pyplot as plt
import sys

sys.path.append('../../..')

from utpnerves.preprocessing import Process

def viz(img):
    plt.imshow(img, cmap='gray')
    plt.axis('off')

images_path = os.path.join('assets', 'images')

list_imgs = list(filter(lambda f:f.endswith('jpg'), os.listdir(images_path)))
list_imgs

['CPD09042021001__20aguja_t_21_4.jpg',
 'CPD09042021001_t_26_4.jpg',
 'CPD09042021002_t_51_85.jpg',
 'CPD09042021002_t_23_9.jpg',
 'CPD09042021001_t_41_9.jpg']

img_path = np.random.choice(list_imgs)
img_array = cv.imread(os.path.join(images_path, img_path), cv.IMREAD_GRAYSCALE)
viz(img_array)

p = Process()
out = p.transform(img_array)
viz(out)

1/1 [==============================] - 0s 219ms/step