1580158
/
batch-face-swap
mirror of https://github.com/kex0/batch-face-swap.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

improved memory management 

fixed corner case exception when dilating low-res mask
pull/17/head
kex0 2023-01-24 02:42:46 +01:00
parent 61998f7bd6
commit 7b2849c741
2 changed files with 17 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.gitignore vendored Normal file
View File

@ -0,0 +1,2 @@
*.pyc

25
scripts/batch_face_swap.py
View File

@ -106,14 +106,16 @@ def findFaces(image, width, height, divider, onlyHorizontal, onlyVertical, file,
imageOriginal = cv2.threshold(imageOriginal,thresh,255,cv2.THRESH_BINARY)[1]
binary_image = cv2.convertScaleAbs(imageOriginal)
# define kernel
kernel = np.ones((int(math.ceil(0.011*height)),int(math.ceil(0.011*height))),'uint8')
dilated = cv2.dilate(binary_image,kernel,iterations=1)
kernel = np.ones((int(math.ceil(0.0045*height)),int(math.ceil(0.0025*height))),'uint8')
dilated = cv2.dilate(dilated,kernel,iterations=1,anchor=(1, -1))
kernel = np.ones((int(math.ceil(0.014*height)),int(math.ceil(0.0025*height))),'uint8')
dilated = cv2.dilate(dilated,kernel,iterations=1,anchor=(-1, 1))
mask = dilated
try:
kernel = np.ones((int(math.ceil(0.011*height)),int(math.ceil(0.011*height))),'uint8')
dilated = cv2.dilate(binary_image,kernel,iterations=1)
kernel = np.ones((int(math.ceil(0.0045*height)),int(math.ceil(0.0025*height))),'uint8')
dilated = cv2.dilate(dilated,kernel,iterations=1,anchor=(1, -1))
kernel = np.ones((int(math.ceil(0.014*height)),int(math.ceil(0.0025*height))),'uint8')
dilated = cv2.dilate(dilated,kernel,iterations=1,anchor=(-1, 1))
mask = dilated
except cv2.error:
mask = dilated
if maskSize != 0:
mask = maskResize(mask, maskSize, height)
@ -198,7 +200,7 @@ def faceSwap(p, masks, image, finishedImages, invertMask, info):
return finishedImages
def generateImages(p, path, searchSubdir, divider, howSplit, saveMask, pathToSave, onlyMask, saveNoFace, overrideDenoising, overrideMaskBlur, invertMask, singleMaskPerImage, countFaces, maskSize, info):
def generateImages(p, path, searchSubdir, viewResults, divider, howSplit, saveMask, pathToSave, onlyMask, saveNoFace, overrideDenoising, overrideMaskBlur, invertMask, singleMaskPerImage, countFaces, maskSize, info):
wasCountFaces = False
finishedImages = []
totalNumberOfFaces = 0
@ -295,6 +297,9 @@ def generateImages(p, path, searchSubdir, divider, howSplit, saveMask, pathToSav
if not onlyMask:
finishedImages = faceSwap(p, masks, image, finishedImages, invertMask, info)
if not viewResults:
finishedImages = []
if wasCountFaces == True:
countFaces = True
@ -589,7 +594,7 @@ class Script(scripts.Script):
all_images = []
divider = int(divider)
finishedImages = generateImages(p, path, searchSubdir, divider, howSplit, saveMask, pathToSave, onlyMask, saveNoFace, overrideDenoising, overrideMaskBlur, invertMask, singleMaskPerImage, countFaces, maskSize, info)
finishedImages = generateImages(p, path, searchSubdir, viewResults, divider, howSplit, saveMask, pathToSave, onlyMask, saveNoFace, overrideDenoising, overrideMaskBlur, invertMask, singleMaskPerImage, countFaces, maskSize, info)
if not viewResults:
finishedImages = []