Распознавание лица на изображении (tkinter + OpenCV)

@Baumanetc · Регистрация: 11.11.2019

Студворк — интернет-сервис помощи студентам

Всем доброго времени суток

Делаю небольшой проект по распознаванию лица на изображении. Суть такова:
1) По нажатию на кнопку вызываю диалог для выбора фото
2) Выбираю изображение и помещаю его на форму в Canvas

Python
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
from tkinter import *
from tkinter import filedialog
from PIL import ImageTk, Image
import cv2.cv2 as cv2
 
 
class Application(Frame):
    def __init__(self, master):
        super(Application, self).__init__(master)
        self.master = master
        self.master.config(bg="#D9D9D9")
        self.create_widgets()
 
    def create_widgets(self):
        image_frame = Frame(self.master)
        image_frame.grid(row=0, column=0, rowspan=6, columnspan=1, sticky=NW)
        self.image_file = LabelFrame(image_frame, text=" original image ")
        self.image_file.grid(padx=5, pady=2)
        self.image_cnvs = Canvas(self.image_file, width=850, height=580)
        self.image_cnvs.grid()
        self.image_file.propagate(False)
 
        self.open_btn = Button(text="open image", width=73, height=3, command=self.open_img)
        self.open_btn.grid(row=1, column=1, columnspan=2, padx=2, sticky=NW)
 
        self.face_btn = Button(text="find face", width=73, height=3, command=self.find_face)
        self.face_btn.grid(row=3, column=1, columnspan=2, padx=2, sticky=NW)
 
    def open_file_name(self):
        filename = filedialog.askopenfilename(title='open')
        return filename
 
    def open_img(self):
        ofn = self.open_file_name()
        img = Image.open(ofn)
        img = ImageTk.PhotoImage(img)
        self.image_cnvs.create_image(10, 10, anchor=NW, image=img)
        self.image_cnvs.image = img
        self.image_cnvs.pack()
 
    def find_face(self):
        face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
        gray = cv2.cvtColor(self.image_cnvs, cv2.COLOR_BGR2GRAY)
        faces = face_cascade.detectMultiScale(gray, 1.3, 5)
        for (x, y, w, h) in faces:
            cv2.rectangle(self.image_cnvs, (x, y), (x + w, y + h), (255, 0, 0), 2)
 
 
def main():
    root = Tk()
    root.title('emotion recognition')
    root.geometry("1400x800")
    root.resizable(FALSE, FALSE)
    app = Application(root)
    root.mainloop()
 
if __name__ == '__main__':
    main()

Нужно так же по нажатию кнопки распознать на этом изображении лицо (или лица, если их несколько). Мой код выше. При нажатии на кнопку получаю ошибки. Вообще это можно как-то сделать? Заранее спасибо за помощь.

@Fudthhh · 18.11.2021, 14:14

Baumanetc, может стоит приложить текст ошибки?

@Baumanetc · 18.11.2021, 22:23 **[ТС]**

Fudthhh, конечно, извините
Вот что я получаю в результате:

Exception in Tkinter callback
Traceback (most recent call last):
File "C:\Python\Python37\lib\tkinter\__init__ .py", line 1705, in __call__
return self.func(*args)
File "D:/Code/Python/test/test7.py", line 45, in find_face
gray = cv2.cvtColor(self.image_cnvs, cv2.COLOR_BGR2GRAY)
cv2.error: OpenCV(4.5.4-dev) :-1: error: (-5:Bad argument) in function 'cvtColor'
> Overload resolution failed:
> - src is not a numpy array, neither a scalar
> - Expected Ptr<cv::UMat> for argument 'src'

Добавлено через 5 часов 58 минут

Python
1
2
3
4
5
6
7
8
9
10
11
    def find_face(self):
        of = self.ofn
        face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
        rec_img = cv2.imread(of)
        gray = cv2.cvtColor(rec_img, cv2.COLOR_BGR2GRAY)
        faces = face_cascade.detectMultiScale(gray, 1.3, 5)
        for (x, y, w, h) in faces:
            cv2.rectangle(rec_img, (x, y), (x + w, y + h), (255, 0, 0), 2)
        cv2.imshow('window', rec_img)
        cv2.waitKey(0)
        cv2.destroyAllWindows()

Этот код выводит изображение, которое я загрузил в Canvas, с распознанным лицом в отдельном окне. А можно его в файл сохранить?
Идея такая:
1) Сохранить изображение с обведенным лицом в файл (например, в папку с проектом)
2) Очистить "старый" Canvas
3) Создать "новый" Canvas с изображением, которое мы сохранили в п. 1)

Добавлено через 2 часа 4 минуты
Как-то так

Python
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
    def find_face(self):
        face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
        face_rec = cv2.imread(self.ofn)
        gray = cv2.cvtColor(face_rec, cv2.COLOR_BGR2GRAY)
        faces = face_cascade.detectMultiScale(gray, 1.3, 5)
        for (x, y, w, h) in faces:
            cv2.rectangle(face_rec, (x, y), (x + w, y + h), (255, 0, 0), 2)
        cv2.imwrite("image-" + time.strftime("%d-%m-%Y-%H-%M-%S") + ".jpg", face_rec)
        file = max(os.listdir(), key=os.path.getctime)
        self.image_cnvs.destroy()
        file_img = Image.open(file)
        file_img = ImageTk.PhotoImage(file_img)
        self.new_image_cnvs = Canvas(self.image_file, width=850, height=580)
        self.new_image_cnvs.grid()
        self.new_image_cnvs.create_image(10, 10, anchor=NW, image=file_img)
        self.new_image_cnvs.image = file_img
        self.new_image_cnvs.pack()

Если кто-то найдет решение лучше, чем мое, поделитесь, пожалуйста. Буду благодарен

@Baumanetc · 19.11.2021, 10:27 **[ТС]**

Python
1
self.image_cnvs.destroy()

лучше заменить на

Python
1
self.image_cnvs.delete("all")

В этом случае не нужно новый Canvas создавать

Новые блоги и статьи Все статьи Все блоги /
SDL3 для Desktop (MinGW): Создаём пустое окно с нуля для 2D-графики на SDL3, Си и C++ 8Observer8 10.03.2026 Содержание блога Финальные проекты на Си и на C++: hello-sdl3-c. zip hello-sdl3-cpp. zip Результат:	Установка CMake и MinGW 13.1 для сборки С и C++ приложений из консоли и из Qt Creator в EXE 8Observer8 10.03.2026 Содержание блога MinGW - это коллекция инструментов для сборки приложений в EXE. CMake - это система сборки приложений. Здесь описаны базовые шаги для старта программирования с помощью CMake и. . .	Как дизайн сайта влияет на конверсию: 7 решений, которые реально повышают заявки Neotwalker 08.03.2026 Многие до сих пор воспринимают дизайн сайта как “красивую оболочку”. На практике всё иначе: дизайн напрямую влияет на то, оставит человек заявку или уйдёт через несколько секунд. Даже если у вас. . .	Модульная разработка через nuget packages DevAlt 07.03.2026 Сложившийся в . Net-среде способ разработки чаще всего предполагает монорепозиторий в котором находятся все исходники. При создании нового решения, мы просто добавляем нужные проекты и имеем. . .
Модульный подход на примере F# DevAlt 06.03.2026 В блоге дяди Боба наткнулся на такое определение: В этой книге («Подход, основанный на вариантах использования») Ивар утверждает, что архитектура программного обеспечения — это структуры,. . .	Управление камерой с помощью скрипта OrbitControls.js на Three.js: Вращение, зум и панорамирование 8Observer8 05.03.2026 Содержание блога Финальная демка в браузере работает на Desktop и мобильных браузерах. Итоговый код: orbit-controls-threejs-js. zip. Сканируйте QR-код на мобильном. Вращайте камеру одним пальцем,. . .	SDL3 для Web (WebAssembly): Синхронизация спрайтов SDL3 и тел Box2D 8Observer8 04.03.2026 Содержание блога Финальная демка в браузере. Итоговый код: finish-sync-physics-sprites-sdl3-c. zip На первой гифке отладочные линии отключены, а на второй включены:. . .	SDL3 для Web (WebAssembly): Идентификация объектов на Box2D v3 - использование userData и событий коллизий 8Observer8 02.03.2026 Содержание блога Финальная демка в браузере. Итоговый код: finish-collision-events-sdl3-c. zip Сканируйте QR-код на мобильном и вы увидите, что появится джойстик для управления главным героем. . . .

@Fudthhh Модератор 2695 / 1601 / 513 Регистрация: 21.02.2017 Сообщений: 4,210 Записей в блоге: 1
	18.11.2021, 14:14
	Baumanetc, может стоит приложить текст ошибки? 0