fix(api): use correct part of the gfpgan output
This commit is contained in:
parent
c6718a3e3b
commit
de8ab6f1c9
|
@ -73,15 +73,15 @@ class CorrectGFPGANStage(BaseStage):
|
||||||
device = worker.get_device()
|
device = worker.get_device()
|
||||||
gfpgan = self.load(server, stage, upscale, device)
|
gfpgan = self.load(server, stage, upscale, device)
|
||||||
|
|
||||||
outputs = [
|
outputs = []
|
||||||
gfpgan.enhance(
|
for source in sources.as_numpy():
|
||||||
|
cropped, restored, result = gfpgan.enhance(
|
||||||
source,
|
source,
|
||||||
has_aligned=False,
|
has_aligned=False,
|
||||||
only_center_face=False,
|
only_center_face=False,
|
||||||
paste_back=True,
|
paste_back=True,
|
||||||
weight=upscale.face_strength,
|
weight=upscale.face_strength,
|
||||||
)
|
)
|
||||||
for source in sources.as_numpy()
|
outputs.append(result)
|
||||||
]
|
|
||||||
|
|
||||||
return StageResult(images=outputs)
|
return StageResult.from_arrays(outputs)
|
||||||
|
|
Loading…
Reference in New Issue