質問編集履歴

tr.pyの内容の一部をsample.pyに統合してみたが、変化がない

2020/06/14 08:46

投稿

mintiamegahard

スコア8

test CHANGED Viewed

File without changes

test CHANGED Viewed

@@ -229,3 +229,181 @@
 ```
 ご回答よろしくお願いいたします。
+###以下、再編集
+##該当のソースコード３（sample.py（統合後））
+```Python
+import tkinter as tk
+from PIL import Image, ImageTk, ImageDraw, ImageFont
+import cv2
+import numpy as np
+from face_recognition import *
+from predict import *
+from create_dataset import *
+root = tk.Tk()
+root.title('sample')
+root.geometry('1000x400+150+10')
+f1 = tk.LabelFrame(root, bd=2, relief="ridge", text="camera")
+f1.pack(side='top', anchor='w', padx=20)
+camera = tk.Canvas(f1, width=600, height=400, bg="white")
+camera.pack(side='left')
+model_path = '../models/20180402-114759'
+pic_path = "../dataset/images"
+dataset_path = '../dataset/emb/faceEmbedding.npy'
+filename = '../dataset/emb/name.txt'
+reload = False
+face_recognition_threshold = 0.85
+print("initializing models...")
+face_detect = Facedetection()
+face_net = facenetEmbedding(model_path)
+if reload:
+    print("refreshing face embeddings...")
+    create_face_embedding(model_path, pic_path, dataset_path, filename)
+print("loading embeddings...")
+dataset_emb, names_list = load_dataset(dataset_path, filename)
+def capture():
+    try:
+        global c, w, h
+        c = cv2.VideoCapture(0)
+        w = c.get(cv2.CAP_PROP_FRAME_WIDTH)
+        h = c.get(cv2.CAP_PROP_FRAME_HEIGHT)
+        pass
+    except:
+        import sys
+        c.release()
+        cv2.destroyAllWindows()
+def up():
+    global img
+    ret, frame = c.read()
+    if ret:
+        result = face_recognition_image_driver(dataset_emb, names_list, face_detect, face_net, frame,
+                                              face_recognition_threshold)
+        def put_text_japanese(img, text, pos):
+            img_PIL = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
+            font_size = 25
+            font = ImageFont.truetype('meiryo', font_size, encoding="utf-8")
+            fillColor = (255, 0, 0)  # 青(0, 0, 255)
+            draw = ImageDraw.Draw(img_PIL)
+            draw.text((pos[0], pos[1] - font_size - 10), text, font=font, fill=fillColor)
+            img = cv2.cvtColor(np.asarray(img_PIL), cv2.COLOR_RGB2BGR)
+            return img
+        frame = put_text_japanese(frame, result, (0, int(h)))
+        img = ImageTk.PhotoImage(Image.fromarray(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)))
+        camera.create_image(w / 2, h / 2, image=img)
+    root.after(1, up)
+  root.after(1, up)
+capture()
+up()
+root.mainloop()
+```
+ご回答よろしくお願いいたします。