lucasrufkahr/Real-ESRGAN
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

use warnings

Browse Source
This commit is contained in:
Xintao
2021-08-09 01:14:54 +08:00
parent 3c6cf5290e
commit 58fea8db69
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
inference_realesrgan.py
View File

@@ -54,9 +54,11 @@ def main():
img = cv2.imread(path, cv2.IMREAD_UNCHANGED) img = cv2.imread(path, cv2.IMREAD_UNCHANGED)
h, w = img.shape[0:2] h, w = img.shape[0:2]
if max(h, w) > 1000 and args.netscale == 4: if max(h, w) > 1000 and args.netscale == 4:
print('WARNING: The input image is large, try X2 model for better performace.') import warnings
warnings.warn('The input image is large, try X2 model for better performace.')
if max(h, w) < 500 and args.netscale == 2: if max(h, w) < 500 and args.netscale == 2:
print('WARNING: The input image is small, try X4 model for better performace.') import warnings
warnings.warn('The input image is small, try X4 model for better performace.')
try: try:
output, img_mode = upsampler.enhance(img, outscale=args.outscale) output, img_mode = upsampler.enhance(img, outscale=args.outscale)