感情検出用のGithub(https://github.com/atulapra/Emotion-detection)からコードが見つかりました。いくつか変更を加え、Tkinterと組み合わせてユーザーとの対話を容易にしたいです。下の画像に示されているこのようなもの。ここで私はtkwindowのコードをいくつか書きました:
win = Tk()
win.title('EMOTIBOT')
win.config(background = "#D9D9D9")
win.resizable(width=FALSE, height=FALSE)
#frame1 for webcamera
frame1 = Frame(win, width=600, bg="black",height=300, padx=10, pady=10,highlightbackground="grey", highlightcolor="black", highlightthickness=5)
frame1.pack(side=LEFT ,fill=Y,padx=10, pady=10)
ここで、元のコードにいくつかの変更を加えました。
if mode == "display":
model.load_weights('model.h5')
# prevents openCL usage and unnecessary logging messages
cv2.ocl.setUseOpenCL(False)
# dictionary which assigns each label an emotion (alphabetical order)
emotion_dict = {0: "Angry", 1: "Disgusted", 2: "Fearful", 3: "Happy", 4: "Neutral", 5: "Sad", 6: "Surprised"}
# start the webcam feed
cap = cv2.VideoCapture(0)
def show_frame():
_, frame = cap.read()
frame = cv2.flip(frame, 1)
#cv2image = cv2.cvtColor(frame, cv2.COLOR_BGR2RGBA)
#while True:
facecasc = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
cv2image = cv2.cvtColor(frame, cv2.COLOR_BGR2RGBA)
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
faces = facecasc.detectMultiScale(cv2image,scaleFactor=1.3, minNeighbors=5)
for (x, y, w, h) in faces:
cv2.rectangle(frame, (x, y-50), (x+w, y+h+10), (255, 0, 0), 2)
roi_gray = gray[y:y + h, x:x + w]
cropped_img = np.expand_dims(np.expand_dims(cv2.resize(roi_gray, (48, 48)), -1), 0)
prediction = model.predict(cropped_img)
maxindex = int(np.argmax(prediction))
cv2.putText(frame, emotion_dict[maxindex], (x+20, y-60), cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 255, 255), 2, cv2.LINE_AA)
cv2.imshow('Video', cv2.resize(frame,(1600,960),interpolation = cv2.INTER_CUBIC))
img = Image.fromarray(cv2image)
imgtk = ImageTk.PhotoImage(image=img)
lmain.imgtk = imgtk
lmain.configure(image=imgtk)
lmain.after(10, show_frame)
#Capture video frames
lmain = Label(frame1)
lmain.grid(row=0, column=0)
show_frame() #Display 2
win.mainloop()
これは2つのウィンドウを開いています。最初のウィンドウ、検出された感情に長方形がないTkウィンドウ。2番目のウィンドウには、検出された感情の長方形が表示されています。最初のtkウィンドウが面に長方形を表示しないのはなぜですか?
私のスクリプトファイルが必要な場合は、ここにあります。(https://drive.google.com/file/d/1m7tLFxHF_hO0VLVb8FgAf-4lJKMxFLxA/view?usp=sharing)
私はOpenCVとTkの両方に不慣れです。私を助けてください。
この記事はインターネットから収集されたものであり、転載の際にはソースを示してください。
侵害の場合は、連絡してください[email protected]
コメントを追加