enhance.py 文件源码

def supic_process(input_path, output_dir, out_width, out_height):
    """ customized api for processing input image """
    try:
        enhancer = NeuralEnhancer(loader=False)
        img = scipy.ndimage.imread(input_path, mode='RGB')

        out = enhancer.process(img)
        out = out.resize((out_width, out_height), PIL.Image.BICUBIC)

        name = hex(int(time.time() * 100000))[2:]
        output_path = os.path.join(output_dir, name + '.png')

        out.save(output_path)
        return output_path
    except Exception as e:
        return '!ERROR' + str(e)