jhj0517
commited on
Commit
·
365e852
1
Parent(s):
6140e4d
Add RealESRGANer wrapper
Browse files
modules/image_restoration/real_esrgan/real_esrgan_inferencer.py
CHANGED
|
@@ -6,9 +6,10 @@ from realesrgan.utils import RealESRGANer
|
|
| 6 |
from realesrgan.archs.srvgg_arch import SRVGGNetCompact
|
| 7 |
|
| 8 |
from modules.utils.paths import *
|
| 9 |
-
from .model_downloader import download_resrgan_model, MODELS_REALESRGAN_URL, MODELS_REALESRGAN_SCALABILITY
|
| 10 |
from modules.utils.image_helper import save_image
|
| 11 |
-
from
|
|
|
|
|
|
|
| 12 |
|
| 13 |
|
| 14 |
class RealESRGANInferencer:
|
|
|
|
| 6 |
from realesrgan.archs.srvgg_arch import SRVGGNetCompact
|
| 7 |
|
| 8 |
from modules.utils.paths import *
|
|
|
|
| 9 |
from modules.utils.image_helper import save_image
|
| 10 |
+
from .model_downloader import download_resrgan_model, MODELS_REALESRGAN_URL, MODELS_REALESRGAN_SCALABILITY
|
| 11 |
+
from wrapper.rrdb_net import RRDBNet
|
| 12 |
+
from wrapper.real_esrganer import RealESRGANer
|
| 13 |
|
| 14 |
|
| 15 |
class RealESRGANInferencer:
|