github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-22 06:23:20 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Don't pass alpha channel of image. Alpha channel may contained if no face was detected in original file and was copied with alpha channel.

Browse source
This commit is contained in:
Alexander Court 2020-05-01 18:40:29 +02:00
commit a03b4506db
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
merger/InteractiveMergerSubprocessor.py
View file

@ -387,7 +387,8 @@ class InteractiveMergerSubprocessor(Subprocessor):
if cur_frame.is_done:
if not cur_frame.is_shown:
if cur_frame.image is None:
image = cv2_imread (cur_frame.output_filepath, verbose=False)
# delete alpha channel
image = cv2_imread (cur_frame.output_filepath, verbose=False)[:,:,0:3]
image_mask = cv2_imread (cur_frame.output_mask_filepath, verbose=False)
if image is None or image_mask is None:
# unable to read? recompute then