diff --git a/inference_realesrgan.py b/inference_realesrgan.py index 819a1b1..09f2815 100644 --- a/inference_realesrgan.py +++ b/inference_realesrgan.py @@ -94,7 +94,7 @@ def main(): _, _, output = face_enhancer.enhance(img, has_aligned=False, only_center_face=False, paste_back=True) else: output, _ = upsampler.enhance(img, outscale=args.outscale) - except Exception as error: + except RuntimeError as error: print('Error', error) print('If you encounter CUDA out of memory, try to set --tile with a smaller number.') else: diff --git a/realesrgan/data/realesrgan_dataset.py b/realesrgan/data/realesrgan_dataset.py index e4c1109..4cf2d9e 100644 --- a/realesrgan/data/realesrgan_dataset.py +++ b/realesrgan/data/realesrgan_dataset.py @@ -92,7 +92,7 @@ class RealESRGANDataset(data.Dataset): while retry > 0: try: img_bytes = self.file_client.get(gt_path, 'gt') - except Exception as e: + except (IOError, OSError) as e: logger = get_root_logger() logger.warn(f'File client error: {e}, remaining retry times: {retry - 1}') # change another file to read diff --git a/realesrgan/utils.py b/realesrgan/utils.py index 802d391..8dad829 100644 --- a/realesrgan/utils.py +++ b/realesrgan/utils.py @@ -128,7 +128,7 @@ class RealESRGANer(): try: with torch.no_grad(): output_tile = self.model(input_tile) - except Exception as error: + except RuntimeError as error: print('Error', error) print(f'\tTile {tile_idx}/{tiles_x * tiles_y}') diff --git a/scripts/generate_meta_info.py b/scripts/generate_meta_info.py index 51d028a..9c3b7a3 100644 --- a/scripts/generate_meta_info.py +++ b/scripts/generate_meta_info.py @@ -14,7 +14,7 @@ def main(args): # read the image once for check, as some images may have errors try: img = cv2.imread(img_path) - except Exception as error: + except (IOError, OSError) as error: print(f'Read {img_path} error: {error}') status = False if img is None: