人脸融合项目求怎么纠错

怪癖行人

1. import numpy as np
   
2. import cv2
   
3. import dlib
   
4. from scipy.spatial import Delaunay
   
5. 
   
6. 
   
7. predictor_model = 'shape_predictor_68_face_landmarks.dat'
   
8. 
   
9. 
   
10. def get_points(image):  # 用 dlib 来得到人脸的特征点
    
11. 
    
12.     face_detector = dlib.get_frontal_face_detector()  # 正向人脸检测器，进行人脸检测，提取人脸外部矩形框
    
13.     face_pose_predictor = dlib.shape_predictor(predictor_model)
    
14.     try:
    
15.         detected_face = face_detector(image, 1)[0]
    
16.     except:
    
17.         print('No face detected in image {}'.format(image))
    
18.     pose_landmarks = face_pose_predictor(image, detected_face)  # 获取landmark
    
19.     points = []
    
20.     for p in pose_landmarks.parts():
    
21.         points.append([p.x, p.y])
    
22. 
    
23.     # 加入四个顶点和四条边的中点
    
24.     x = image.shape[1] - 1
    
25.     y = image.shape[0] - 1
    
26.     points.append([0, 0])
    
27.     points.append([x // 2, 0])
    
28.     points.append([x, 0])
    
29.     points.append([x, y // 2])
    
30.     points.append([x, y])
    
31.     points.append([x // 2, y])
    
32.     points.append([0, y])
    
33.     points.append([0, y // 2])
    
34. 
    
35.     return np.array(points)
    
36. 
    
37. 
    
38. def get_triangles(points):  #  在特征点上使用 Delaunay 三角剖分，将点集连接成一定大小的三角形，且分配要相对合理，才能呈现出漂亮的三角化
    
39.     return Delaunay(points).simplices
    
40. 
    
41. 
    
42. def affine_transform(input_image, input_triangle, output_triangle, size):  # 对人脸进行仿射变换，确定位置
    
43.     warp_matrix = cv2.getAffineTransform(
    
44.         np.float32(input_triangle), np.float32(output_triangle))
    
45.     output_image = cv2.warpAffine(input_image, warp_matrix, (size[0], size[1]), None,
    
46.                                   flags=cv2.INTER_LINEAR, borderMode=cv2.BORDER_REFLECT_101)
    
47.     return output_image
    
48. 
    
49. 
    
50. def morph_triangle(img1, img2, img, tri1, tri2, tri, alpha):  # 三角形变形，Alpha 混合
    
51.     # 计算三角形的边界框
    
52.     rect1 = cv2.boundingRect(np.float32([tri1]))  # 寻找tri1的左上角坐标，和tri1的长和宽
    
53.     rect2 = cv2.boundingRect(np.float32([tri2]))
    
54.     rect = cv2.boundingRect(np.float32([tri]))
    
55. 
    
56.     tri_rect1 = []
    
57.     tri_rect2 = []
    
58.     tri_rect_warped = []
    
59. 
    
60.     for i in range(0, 3):
    
61.         tri_rect_warped.append(
    
62.             ((tri[i][0] - rect[0]), (tri[i][1] - rect[1])))
    
63.         tri_rect1.append(
    
64.             ((tri1[i][0] - rect1[0]), (tri1[i][1] - rect1[1])))
    
65.         tri_rect2.append(
    
66.             ((tri2[i][0] - rect2[0]), (tri2[i][1] - rect2[1])))
    
67. 
    
68.     # 在边界框内进行仿射变换
    
69.     img1_rect = img1[rect1[1]:rect1[1] +
    
70.                      rect1[3], rect1[0]:rect1[0] + rect1[2]]
    
71.     img2_rect = img2[rect2[1]:rect2[1] +
    
72.                      rect2[3], rect2[0]:rect2[0] + rect2[2]]
    
73. 
    
74.     size = (rect[2], rect[3])
    
75.     warped_img1 = affine_transform(
    
76.         img1_rect, tri_rect1, tri_rect_warped, size)
    
77.     warped_img2 = affine_transform(
    
78.         img2_rect, tri_rect2, tri_rect_warped, size)
    
79. 
    
80.     # 加权求和
    
81.     img_rect = (1.0 - alpha) * warped_img1 + alpha * warped_img2
    
82. 
    
83.     # 生成模板
    
84.     mask = np.zeros((rect[3], rect[2], 3), dtype=np.float32)
    
85.     cv2.fillConvexPoly(mask, np.int32(tri_rect_warped), (1.0, 1.0, 1.0), 16, 0)
    
86. 
    
87.     # 应用模板
    
88.     img[rect[1]:rect[1] + rect[3], rect[0]:rect[0] + rect[2]] = \
    
89.         img[rect[1]:rect[1] + rect[3], rect[0]:rect[0] +
    
90.             rect[2]] * (1 - mask) + img_rect * mask
    
91. 
    
92. 
    
93. def morph_faces(filename1, filename2, alpha=0.5):  # 融合图片
    
94.     img1 = cv2.imread(filename1)
    
95.     img2 = cv2.imread(filename2)
    
96.     img2 = cv2.resize(img2,(img1.shape[1],img1.shape[0]),interpolation=cv2.INTER_CUBIC)
    
97.     print('img1.shape',img1.shape)
    
98.     print('img2.shape',img2.shape)
    
99. 
    
100.     points1 = get_points(img1)
     
101.     print('pionts1:',len(points1),points1)
     
102.     points2 = get_points(img2)
     
103.     points = (1 - alpha) * np.array(points1) + alpha * np.array(points2)
     
104.     import pandas as pd
     
105.     p = pd.DataFrame(points)
     
106.     p.to_csv('./1.csv')
     
107. 
     
108.     img1 = np.float32(img1)
     
109.     img2 = np.float32(img2)
     
110.     img_morphed = np.zeros(img1.shape, dtype=img1.dtype)
     
111. 
     
112.     triangles = get_triangles(points)
     
113.     for i in triangles:
     
114.         x = i[0]
     
115.         y = i[1]
     
116.         z = i[2]
     
117. 
     
118.         tri1 = [points1[x], points1[y], points1[z]]
     
119.         tri2 = [points2[x], points2[y], points2[z]]
     
120.         tri = [points[x], points[y], points[z]]
     
121.         morph_triangle(img1, img2, img_morphed, tri1, tri2, tri, alpha)
     
122. 
     
123.     return np.uint8(img_morphed)
     
124. 
     
125. 
     
126. def main(file1,file2,alpha):
     
127.     try:
     
128.      alpha = float(alpha)
     
129.     except:
     
130.         alpha = 0.5
     
131.     img_morphed = morph_faces(file1, file2, alpha)
     
132.     output_file = '{}_{}_{}.jpg'.format(
     
133.         file1.split('.')[0][-2:], file2.split('.')[0][-1:], alpha)
     
134.     cv2.imwrite(output_file, img_morphed)
     
135.     return output_file
     
136. 
     
137. 
     

复制代码

1. from PIL import Image, ImageTk
   
2. from tkinter.filedialog import askopenfilename
   
3. from tkinter import *
   
4. import PIL
   
5. from face_morhper import main
   
6. 
   
7. root = Tk()
   
8. root.title('人脸融合软件')
   
9. root.geometry('1200x500')
   
10. 
    
11. decoration = PIL.Image.open('zyf.jpg').resize((1200, 500))
    
12. render = ImageTk.PhotoImage(decoration)
    
13. img = Label(image=render)
    
14. img.image = render
    
15. img.place(x=0, y=0)
    
16. 
    
17. global path1_, path2_, rate, seg_img_path
    
18. 
    
19. 
    
20. # 原图1展示
    
21. def show_original1_pic():
    
22.     global path1_
    
23.     path1_ = askopenfilename(title='选择文件')
    
24.     print(path1_)
    
25.     Img = PIL.Image.open(r'{}'.format(path1_))
    
26.     Img = Img.resize((270,270),PIL.Image.ANTIALIAS)   # 调整图片大小至256x256
    
27.     img_png_original = ImageTk.PhotoImage(Img)
    
28.     label_Img_original1.config(image=img_png_original)
    
29.     label_Img_original1.image = img_png_original  # keep a reference
    
30.     cv_orinial1.create_image(5, 5,anchor='nw', image=img_png_original)
    
31. 
    
32. 
    
33. # 原图2展示
    
34. def show_original2_pic():
    
35.     global path2_
    
36.     path2_ = askopenfilename(title='选择文件')
    
37.     print(path2_)
    
38.     Img = PIL.Image.open(r'{}'.format(path2_))
    
39.     Img = Img.resize((270,270),PIL.Image.ANTIALIAS)   # 调整图片大小至256x256
    
40.     img_png_original = ImageTk.PhotoImage(Img)
    
41.     label_Img_original2.config(image=img_png_original)
    
42.     label_Img_original2.image = img_png_original  # keep a reference
    
43.     cv_orinial2.create_image(5, 5,anchor='nw', image=img_png_original)
    
44. 
    
45. 
    
46. # 人脸融合效果展示
    
47. def show_morpher_pic():
    
48.     global path1_,seg_img_path,path2_
    
49.     print(entry.get())
    
50.     mor_img_path = main(path1_,path2_,entry.get())
    
51.     Img = PIL.Image.open(r'{}'.format(mor_img_path))
    
52.     Img = Img.resize((270, 270), PIL.Image.ANTIALIAS)  # 调整图片大小至256x256
    
53.     img_png_seg = ImageTk.PhotoImage(Img)
    
54.     label_Img_seg.config(image=img_png_seg)
    
55.     label_Img_seg.image = img_png_seg  # keep a reference
    
56. 
    
57. 
    
58. def quit():
    
59.     root.destroy()
    
60. 
    
61. 
    
62. # 原图1的展示
    
63. Button(root, text = "打开图片1", command = show_original1_pic).place(x=50,y=120)
    
64. # 原图2的展示
    
65. Button(root, text = "打开图片2", command = show_original2_pic).place(x=50,y=200)
    
66. # 进行提取结果的展示
    
67. Button(root, text = "人脸融合", command = show_morpher_pic).place(x=50,y=280)
    
68. 
    
69. Button(root, text = "退出软件", command = quit).place(x=900,y=40)
    
70. 
    
71. Label(root,text = "融合系数",font=10).place(x=50,y=10)
    
72. entry = Entry(root)
    
73. entry.place(x=130,y=10)
    
74. 
    
75. 
    
76. Label(root,text = "图片1",font=10).place(x=280,y=120)
    
77. cv_orinial1 = Canvas(root,bg = 'white',width=270,height=270)
    
78. cv_orinial1.create_rectangle(8,8,260,260,width=1,outline='red')
    
79. cv_orinial1.place(x=180,y=150)
    
80. label_Img_original1 = Label(root)
    
81. label_Img_original1.place(x=180,y=150)
    
82. 
    
83. 
    
84. Label(root,text="图片2",font=10).place(x=600,y=120)
    
85. cv_orinial2 = Canvas(root,bg = 'white',width=270,height=270)
    
86. cv_orinial2.create_rectangle(8,8,260,260,width=1,outline='red')
    
87. cv_orinial2.place(x=500,y=150)
    
88. label_Img_original2 = Label(root)
    
89. label_Img_original2.place(x=500,y=150)
    
90. 
    
91. Label(root, text="融合效果", font=10).place(x=920,y=120)
    
92. cv_seg = Canvas(root, bg='white', width=270,height=270)
    
93. cv_seg.create_rectangle(8,8,260,260,width=1,outline='red')
    
94. cv_seg.place(x=820,y=150)
    
95. label_Img_seg = Label(root)
    
96. label_Img_seg.place(x=820,y=150)
    
97. 
    
98. 
    
99. root.mainloop()

复制代码

报错：
Exception in Tkinter callback
Traceback (most recent call last):
  File "C:\Users\12160\AppData\Local\Programs\Python\Python39\lib\tkinter\__init__.py", line 1892, in __call__
    return self.func(*args)
  File "E:\人脸合成\interface.py", line 50, in show_morpher_pic
    mor_img_path = main(path1_,path2_,entry.get())
  File "E:\人脸合成\face_morhper.py", line 131, in main
    img_morphed = morph_faces(file1, file2, alpha)
  File "E:\人脸合成\face_morhper.py", line 96, in morph_faces
    img2 = cv2.resize(img2,(img1.shape[1],img1.shape[0]),interpolation=cv2.INTER_CUBIC)
AttributeError: 'NoneType' object has no attribute 'shape'

nahongyan1997

wocao
吊！！