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