Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-09-01 19:49:03 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix image_portrait_enhancement rgb channel bug (#740)

Browse Source 
Co-authored-by: Starsky Wong <wangsunping@rayvision.com>
This commit is contained in:
Starsky Wong
2024-03-04 10:45:28 +08:00
committed by GitHub
parent a0fb7e6fd8
commit 2d2d9e4fe9
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
modelscope/pipelines/cv/image_portrait_enhancement_pipeline.py
View File

@@ -173,11 +173,13 @@ class ImagePortraitEnhancementPipeline(Pipeline):
def preprocess(self, input: Input) -> Dict[str, Any]:
img = LoadImage.convert_to_ndarray(input)
img_sr = img
if self.use_sr:
img_sr = self.sr_process(img)
img = cv2.resize(img, img_sr.shape[:2][::-1])
img = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
else:
img = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
img_sr = img.copy()
result = {'img': img, 'img_sr': img_sr}
return result
@@ -200,6 +202,9 @@ class ImagePortraitEnhancementPipeline(Pipeline):
of, of_112, tfm_inv = warp_and_crop_face(
img, facial5points, crop_size=(self.size, self.size))
of = of[..., ::-1].copy() # BGR->RGB
of_112 = of_112[..., ::-1].copy() # BGR->RGB
# detect orig face quality
fq_o, fea_o = self.eqface.get_face_quality(of_112)
if fq_o < self.fqa_thres: