init

APDrawingGAN2/preprocess/get_partmask.py

@@ -66,10 +66,10 @@ def get_68lm(imgfile,savepath5,savepath68, detector, predictor):
 
 		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[36]+landmarks[39])/2
+		print(int(lm[0]), int(lm[1]), file=ff)
+		lm = (landmarks[45]+landmarks[42])/2
+		print(int(lm[0]), int(lm[1]), file=ff)
 		lm = landmarks[30]
 		print(lm[0], lm[1], file=ff)
 		lm = landmarks[48]