fastai predict array

predict_array.py

from fastai.vision import Image,pil2tensor
import cv2

def array2tensor(x):
    """ Return an tensor image from cv2 array """
    x = cv2.cvtColor(x,cv2.COLOR_BGR2RGB)
    return Image(pil2tensor(x,np.float32).div_(255))
    
    
### USAGE
learn = load_learner(model_path,model_name)
# insert a new function to learner
learn.predict_array = types.MethodType( predict_array, learn )

img = cv2.imread([IMAGE FILE NAME])
learn.predict_array(img)