src/util/image.py

from PIL import Image 
import numpy as np
import requests

def load_image_url(url, required_size = (224,224), image_type = 'array'):
    print(f'downloading.. {url}, type: {image_type}')
    img = Image.open(requests.get(url, stream=True).raw)
    img = Image.fromarray(np.array(img))
    if required_size is not None:
        img = img.resize(required_size)
    if image_type == 'array':
        img = (np.expand_dims(np.array(img), 0)/255).astype(np.float32)
    return img

def load_image_path(path, required_size = (224,224), image_type = 'array'):
    img = Image.open(path)
    img = Image.fromarray(np.array(img))
    if required_size is not None:
        img = img.resize(required_size)
    if image_type == 'array':
        img = (np.expand_dims(np.array(img), 0)/255).astype(np.float32)
    return img