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

Merge pull request #80 from faceshiftlabs/fix/mask-editor-padding

Browse source 
Fix padding on mask_editor
This commit is contained in:
Jeremy Hummel 2019-09-26 11:45:50 -07:00 • committed by GitHub
commit dadd579bf9
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
mainscripts/MaskEditorTool.py
View file

@ -29,7 +29,8 @@ class MaskEditor:
# self.img = cv2.resize(img, (256,256)) # self.img = cv2.resize(img, (256,256))
# raise Exception ("MaskEditor does not support image size != 256x256") # raise Exception ("MaskEditor does not support image size != 256x256")
ph, pw = h // 4, w // 4 #pad wh # ph, pw = h // 4, w // 4 #pad wh
ph, pw = 64, 64
self.prev_images = prev_images self.prev_images = prev_images
self.next_images = next_images self.next_images = next_images
@ -267,7 +268,7 @@ class MaskEditor:
preview_images = np.concatenate ( preview_images, axis=1 ) preview_images = np.concatenate ( preview_images, axis=1 )
left_pad = sw // 2 - len(prev_images) * prw - prw // 2 left_pad = sw // 2 - len(prev_images) * prw - prw // 2
right_pad = sw // 2 - len(next_images) * prw - prw // 2 right_pad = (sw + 1) // 2 - len(next_images) * prw - prw // 2
preview_images = np.concatenate ([np.zeros ( (preview_images.shape[0], left_pad, preview_images.shape[2]) ), preview_images = np.concatenate ([np.zeros ( (preview_images.shape[0], left_pad, preview_images.shape[2]) ),
preview_images, preview_images,