Source code phần mềm phát hiện không đeo khẩu trang bằng Python

Source code phần mềm phát hiện không đeo khẩu trang bằng Python

Bước 1: Phát hiện khuôn mặt trong ảnh Đầu tiên, phần mềm sẽ phát hiện khuôn mặt trong ảnh, bất kể người đó có đeo khẩu trang hay không. Sau đó, khuôn mặt sẽ được cắt (crop) để chuẩn bị cho bước tiếp theo. Ở bước này, kỹ thuật phát hiện khuôn mặt được sử dụng là LBP cascade. Mặc dù đây là một phương pháp cũ, nhưng vẫn mang lại hiệu quả tốt trong việc nhận diện khuôn mặt.

Bước 2: Phân loại khuôn mặt để xác định việc đeo khẩu trang Sau khi phát hiện và cắt khuôn mặt, bước này sẽ phân loại các khuôn mặt để xác định tình trạng đeo khẩu trang, bao gồm:

• Mask: Có đeo khẩu trang
• No mask: Không đeo khẩu trang
• Wrong: Đeo khẩu trang sai cách (chừa mũi hoặc miệng)
• Hand: Dùng tay che mặt hoặc khẩu trang
• Nothing: Trường hợp nhận diện nhầm khu vực không có khuôn mặt

Chỉ có trạng thái “Mask” là hợp lệ. Những trường hợp khác cần được nhắc nhở đeo khẩu trang đúng cách. Để phân loại các trạng thái này, phần mềm sử dụng mô hình MobileNet để huấn luyện và dự đoán.

Project này sử dụng các thư viện sau:

• OpenCV
• Tensorflow

Cài đặt

•      Tải source code về
• Cài đặt Python 3.7.3 x64
• Cài package cần thiết bằng lệnh: pip3 install -r requirements.txt
• Chạy file FaceMaskDetector.py

Nếu thành công thì chương trình sẽ hiện lên hình ảnh từ webcam và vẽ khuôn mặt lên đó. Các bạn có thể code thêm để chương trình hoàn chỉnh hơn như phát ra âm thanh nhắc nhở, ghi nhận người không đeo khẩu trang,…