v2.1 - Face correction (GFPGAN) and Upscaling (RealESRGAN)

This commit is contained in:
cmdr2
2022-09-08 21:20:27 +05:30
parent 22089c528a
commit 253e75c747
6 changed files with 531 additions and 46 deletions

View File

@ -17,6 +17,9 @@ class Request:
turbo: bool = True
use_cpu: bool = False
use_full_precision: bool = False
use_face_correction: str = None # or "GFPGANv1.3"
use_upscale: str = None # or "RealESRGAN_x4plus" or "RealESRGAN_x4plus_anime_6B"
show_only_filtered_image: bool = False
def to_string(self):
return f'''
@ -30,7 +33,10 @@ class Request:
save_to_disk_path: {self.save_to_disk_path}
turbo: {self.turbo}
use_cpu: {self.use_cpu}
use_full_precision: {self.use_full_precision}'''
use_full_precision: {self.use_full_precision}
use_face_correction: {self.use_face_correction}
use_upscale: {self.use_upscale}
show_only_filtered_image: {self.show_only_filtered_image}'''
class Image:
data: str # base64