jhj0517 commited on
Commit
2d217e7
1 Parent(s): 922c507

Use util function

Browse files
modules/live_portrait/live_portrait_inferencer.py CHANGED
@@ -205,11 +205,10 @@ class LivePortraitInferencer:
205
  crop_with_fullsize = cv2.warpAffine(crop_out, psi.crop_trans_m, get_rgb_size(psi.src_rgb), cv2.INTER_LINEAR)
206
  out = np.clip(psi.mask_ori * crop_with_fullsize + (1 - psi.mask_ori) * psi.src_rgb, 0, 255).astype(np.uint8)
207
 
208
- cropped_img, out_img = Image.fromarray(crop_out), Image.fromarray(out)
209
  temp_out_img_path, out_img_path = get_auto_incremental_file_path(TEMP_DIR, "png"), get_auto_incremental_file_path(OUTPUTS_DIR, "png")
 
 
210
 
211
- cropped_img.save(temp_out_img_path, compress_level=1, format="png")
212
- out_img.save(out_img_path, compress_level=1, format="png")
213
  new_editor_link.append(es)
214
 
215
  return out
 
205
  crop_with_fullsize = cv2.warpAffine(crop_out, psi.crop_trans_m, get_rgb_size(psi.src_rgb), cv2.INTER_LINEAR)
206
  out = np.clip(psi.mask_ori * crop_with_fullsize + (1 - psi.mask_ori) * psi.src_rgb, 0, 255).astype(np.uint8)
207
 
 
208
  temp_out_img_path, out_img_path = get_auto_incremental_file_path(TEMP_DIR, "png"), get_auto_incremental_file_path(OUTPUTS_DIR, "png")
209
+ save_image(numpy_array=crop_out, output_path=temp_out_img_path)
210
+ save_image(numpy_array=out, output_path=out_img_path)
211
 
 
 
212
  new_editor_link.append(es)
213
 
214
  return out