detectron2 web app low memory image transforms

img_transforms.py

import PIL
from PIL import Image

# adapted from: https://github.com/jantic/DeOldify/blob/master/deoldify/filters.py

def _scale_to_square(orig, targ):
  targ_sz = (targ, targ)
  return orig.resize(targ_sz, resample=PIL.Image.BILINEAR)


def _unsquare(image, orig):
  targ_sz = orig.size
  image = image.resize(targ_sz, resample=PIL.Image.BILINEAR)
  return images