def load_imgs(path, size):
files = glob.glob(os.path.join(path, "*"))
imgs = []
for file in files:
img = cv2.imread(file)
img = DataGenerator.pre_process(img, size)
imgs.append(img)
return files, imgs
def split_path(list_imgs, size):
path_imgs = []
imgs = []
for file in list_imgs:
img = cv2.imread(file['path'])
img = DataGenerator.pre_process(img, size)
imgs.append(img)
path_imgs.append(file['path'])
return path_imgs, imgs
def get_test_imgs(path, size):
path_imgs = []
for root, dir, files in os.walk(path):
for file in files:
file_path = os.path.join(root, file)
path_imgs.append(file_path)
imgs = []
for file in path_imgs:
img = cv2.imread(file)
img = DataGenerator.pre_process(img, size)
imgs.append(img)
return path_imgs, imgs
def pre_process(img_path):
img = cv2.imread(img_path)
img = DataGenerator.pre_process(img, size=(input_shape[1], input_shape[0]))
return img
