init

APDrawingGAN2/preprocess/get_partmask.py

@@ -51,19 +51,34 @@ def shape_to_np(shape, dtype="int"):
 	# return the list of (x, y)-coordinates
 	return coords
 
-def get_68lm(imgfile,savepath, detector, predictor):
+def get_68lm(imgfile,savepath5,savepath68, detector, predictor):
 	image = cv2.imread(imgfile)
 	rgbImg = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
 	rects = detector(rgbImg, 1)
 	for (i, rect) in enumerate(rects):
 		landmarks = predictor(rgbImg, rect)
 		landmarks = shape_to_np(landmarks)
-		f = open(savepath,'w')
+		f = open(savepath68,'w')
 		for i in range(len(landmarks)):
 			lm = landmarks[i]
 			print(lm[0], lm[1], file=f)
 		f.close()
 
+		ff = open(savepath5, 'w')
+
+		lm = landmarks[36]
+		print(lm[0], lm[1], file=ff)
+		lm = landmarks[45]
+		print(lm[0], lm[1], file=ff)
+		lm = landmarks[30]
+		print(lm[0], lm[1], file=ff)
+		lm = landmarks[48]
+		print(lm[0], lm[1], file=ff)
+		lm = landmarks[54]
+		print(lm[0], lm[1], file=ff)
+
+		ff.close()
+
 def get_partmask(imgfile,part,lmpath,savefile):
 	img = cv2.imread(imgfile)
 	mask = np.zeros(img.shape, np.uint8)

app.py

@@ -139,14 +139,15 @@ def run(
     shutil.copyfile(mask.name, os.path.join(opt.bg_dir, name+'.png'))
 
     imgfile = os.path.join(opt.dataroot, fullname)
-    lmfile = os.path.join(opt.lm_dir, name+'.txt')
+    lmfile5 = os.path.join(opt.lm_dir, name+'.txt')
+    lmfile68 = os.path.join(opt.lm_dir, name + '_68.txt')
     # 预处理数据
-    get_68lm(imgfile, lmfile, detector, predictor)
+    get_68lm(imgfile, lmfile5, lmfile68, detector, predictor)
 
     imgs = []
     for part in ['eyel', 'eyer', 'nose', 'mouth']:
         savepath = os.path.join(opt.cmask_dir + part, name+'.png')
-        get_partmask(imgfile, part, lmfile, savepath)
+        get_partmask(imgfile, part, lmfile68, savepath)
         #imgs.append(savepath)
 
     data_loader = CreateDataLoader(opt)