Почему opencv создает 2 windows? - PullRequest
2 голосов
/ 27 апреля 2020

Найден код от Github (https://github.com/atulapra/Emotion-detection) для обнаружения эмоций. Я хочу внести в него некоторые изменения и объединить их с Tkinter для облегчения взаимодействия с пользователем. Нечто подобное показано на рисунке ниже. enter image description here Здесь я написал код для окна tk:

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 windows. Первое окно, окно Tk без прямоугольников на обнаруженные эмоции. Второе окно показывает прямоугольники на обнаруженные эмоции. Почему в первом окне tk не отображаются прямоугольники на гранях? enter image description here

если вам нужен мой файл скрипта, он здесь. (https://drive.google.com/file/d/1m7tLFxHF_hO0VLVb8FgAf-4lJKMxFLxA/view?usp=sharing)

Я новичок в OpenCV и Tk. Пожалуйста, помогите мне.

1 Ответ

2 голосов
/ 27 апреля 2020

Чтобы обобщить наше обсуждение в ответ , который может быть принят :

  1. Удалить cv2.imshow, чтобы избавиться от второго окна
  2. Использовать img = Image.fromarray(cv2.cvtColor(frame, cv2.COLOR_BGR2RGBA)) для отображения рамки с обнаруженным лицом правильными цветами.
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...