v/GFPGAN
1
0
Fork 0
mirror of https://github.com/TencentARC/GFPGAN.git synced 2025-05-15 14:50:11 -07:00
Code Issues Projects Releases Wiki Activity

fix pylint

Browse Source
This commit is contained in:
Xintao 2022-09-16 19:32:42 +08:00
parent bc5a5deb95
commit 2f46d95254
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
cog_predict.py
View File

@ -64,13 +64,13 @@ class Predictor(BasePredictor):
self.current_version = 'v1.4' self.current_version = 'v1.4'
def predict( def predict(
self, self,
img: Path = Input(description='Input'), img: Path = Input(description='Input'),
version: str = Input( version: str = Input(
description='GFPGAN version. v1.3: better quality. v1.4: more details and better identity.', description='GFPGAN version. v1.3: better quality. v1.4: more details and better identity.',
choices=['v1.2', 'v1.3', 'v1.4', 'RestoreFormer'], choices=['v1.2', 'v1.3', 'v1.4', 'RestoreFormer'],
default='v1.4'), default='v1.4'),
scale: float = Input(description='Rescaling factor', default=2), scale: float = Input(description='Rescaling factor', default=2),
) -> Path: ) -> Path:
weight = 0.5 weight = 0.5
print(img, version, scale, weight) print(img, version, scale, weight)