thiagoh/simple-predictor

1import os2from pathlib import Path3from fastcore.transform import Transform4from fastai.vision.widgets import *5from fastai.vision import *6from fastai.vision.all import *7from fastai.learner import load_learner8import PIL9 10class GrayscaleTransform(Transform):11    def __init__(self):12      pass13    def encodes(self, img: PILImage):14      return PIL.ImageOps.grayscale(img)15 16def get_fingers(fingers_path):17    print('fingers path', fingers_path)18    return []19 20def get_finger_label(finger_path: Path):21    print('finger path', finger_path)22    return ''23 24def local_load_learner(model_filename):25  current_path = Path(os.getcwd())26  fingers_model_path = Path(current_path/'models'/model_filename)27  print(f'Model file {model_filename} exists? ', fingers_model_path.exists())28  print(f'Stats ', os.lstat(fingers_model_path.as_posix()))29  finger_learn_inf: Learner = load_learner(fingers_model_path)30  print(f'Model ({model_filename}) successfully loaded: {finger_learn_inf}')31  return finger_learn_inf32 33def setup_model() -> Tuple[Learner, Learner]:34  bear_learner: Learner = local_load_learner('export.pkl')35  finger_learner: Learner = local_load_learner('fingers-2023-07-01-20-46-08.pkl')36  return (finger_learner, bear_learner)37