zhiweili
commited on
Commit
·
0ff5d61
1
Parent(s):
98f53ae
change scale
Browse files- enhance_utils.py +1 -1
enhance_utils.py
CHANGED
@@ -34,7 +34,7 @@ os.makedirs('output', exist_ok=True)
|
|
34 |
model = SRVGGNetCompact(num_in_ch=3, num_out_ch=3, num_feat=64, num_conv=32, upscale=4, act_type='prelu')
|
35 |
model_path = 'realesr-general-x4v3.pth'
|
36 |
half = True if torch.cuda.is_available() else False
|
37 |
-
upsampler = RealESRGANer(scale=
|
38 |
|
39 |
face_enhancer = GFPGANer(model_path='GFPGANv1.4.pth', upscale=2, arch='clean', channel_multiplier=2)
|
40 |
|
|
|
34 |
model = SRVGGNetCompact(num_in_ch=3, num_out_ch=3, num_feat=64, num_conv=32, upscale=4, act_type='prelu')
|
35 |
model_path = 'realesr-general-x4v3.pth'
|
36 |
half = True if torch.cuda.is_available() else False
|
37 |
+
upsampler = RealESRGANer(scale=4, model_path=model_path, model=model, tile=0, tile_pad=10, pre_pad=0, half=half)
|
38 |
|
39 |
face_enhancer = GFPGANer(model_path='GFPGANv1.4.pth', upscale=2, arch='clean', channel_multiplier=2)
|
40 |
|