From ec8516d550a1f7cfdec3a0abe21319ef776a0ef7 Mon Sep 17 00:00:00 2001 From: vexaroniter <52722847+vexaroniter@users.noreply.github.com> Date: Wed, 9 Dec 2020 05:05:01 +0400 Subject: [PATCH] Work without deprecated SciPy methods (Py3.9+) -In the last version SciPy some methods have been deprecated. -Methods have been replaced to PIL* & imageio* --------------------_______More details_______-------------------- -The read method has been replaced from "scipy.ndimage.imread(filename, mode='RGB')" to "imageio.imread(filename, as_gray=False, pilmode="RGB")" -The return buffer method has been replaced from "scipy.misc.toimage(output, cmin=0, cmax=255)" to "PIL.Image.fromarray((output).astype('uint8'), mode='RGB')" --- enhance.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/enhance.py b/enhance.py index becb851..b98d332 100755 --- a/enhance.py +++ b/enhance.py @@ -20,11 +20,13 @@ import os import sys import bz2 +import PIL import glob import math import time import pickle import random +import imageio import argparse import itertools import threading @@ -568,7 +570,7 @@ def process(self, original): for i in range(3): output[:,:,i] = self.match_histograms(output[:,:,i], original[:,:,i]) - return scipy.misc.toimage(output, cmin=0, cmax=255) + return PIL.Image.fromarray((output).astype('uint8'), mode='RGB') if __name__ == "__main__": @@ -580,7 +582,7 @@ def process(self, original): enhancer = NeuralEnhancer(loader=False) for filename in args.files: print(filename, end=' ') - img = scipy.ndimage.imread(filename, mode='RGB') + img = imageio.imread(filename, as_gray=False, pilmode="RGB") out = enhancer.process(img) out.save(os.path.splitext(filename)[0]+'_ne%ix.png' % args.zoom) print(flush=True)