faceswap-2/faceswap.py at master · zanDaYa/faceswap-2 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78

import sys
import os
import cv2
from facerepl import *

from facelib.facepp import API, File
from facelib.faceinfo import FaceInfo, FacePose, FaceLandmark
from facelib.facerank import FaceRank
from facelib.faceposebin import FacePoseBin, FacePoseBinDataFile
from facelib.faceaws import *

DEFAULT_NUM_TOP = 6
DEFAULT_OUT_PREFIX = 'tmp_'
DEFAULT_DATAFILE_PATH = 'datafile.txt'

API_KEY = 'bf7eae5ed9cf280218450523049d5f94'
API_SECRET = 'o6SeKJTnaoczTb-j6PBEGXvkiVz2hp71'
api = API(API_KEY, API_SECRET)

def faceswap(img_name):
	img_parsed_name = (img_name.split('/')[-1]).split('.')[-2]
	print img_parsed_name
	img = cv2.imread(img_name)
	dummybin = FacePoseBin()
	datafile = FacePoseBinDataFile(DEFAULT_DATAFILE_PATH)

	rst = api.detection.detect(img = File(img_name), attribute='pose')
	img_width = rst['img_width']
	img_height = rst['img_height']
	face = rst['face'][0]
	face_id = face['face_id']
	landmark = api.detection.landmark(face_id = face_id, type='25p')
	pose = face['attribute']['pose']
	face_pose = FacePose(pose)
	face_landmark = FaceLandmark(landmark['result'][0]['landmark'], img_width, img_height)

	cur_face_info = FaceInfo()
	cur_face_info.pose = face_pose
	cur_face_info.landmark = face_landmark

	face_posebin_id = dummybin.get_pose_bin_id(face_pose.yaw, face_pose.pitch)
	print face_id
	print face_pose
	print face_landmark
	print 'current face has posebin id %d' % face_posebin_id
	cur_face_info.pose_bin_id = face_posebin_id

	rank = FaceRank(cur_face_info, datafile)
	faceinfos = rank.rank()

	print "found and sorted %d faces" % len(faceinfos)

	ret_names = []
	src_names = []
	for faceinfo in faceinfos[0:DEFAULT_NUM_TOP]:
		#Here to request the remote image
		src_name = get_image(faceinfo.pose_bin_id, faceinfo.name)
		#src_name = "%s/%d/%s.png" % (posebin.path, faceinfo.pose_bin_id, faceinfo.name)
		dst_name = img_name
		swap_img = facecloning.faceclone(src_name, dst_name)
		swap_name = DEFAULT_OUT_PREFIX+img_parsed_name+'_'+faceinfo.name+'.png'
		cv2.imwrite(swap_name, swap_img)

		ret_names.append(swap_name)
		src_names.append(src_name)
	for src in src_names:
		os.remove(src)
	return ret_names


if __name__ == '__main__':
	if len(sys.argv) <= 1:
		print "usage: %s <image>" % sys.argv[0]
		exit(1)

	print "swapping face %s ..." % sys.argv[1]
	print faceswap(sys.argv[1])