lltlien / eyeglasses-detection

The final project for my class Computer Vision : Eyeglasses Detection using deepface, Mobilenetv2 and Resnet101v2

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Trường Đại học Công nghệ Thông tin | University of Information Technology

NHẬP MÔN THỊ GIÁC MÁY TÍNH

THÀNH VIÊN NHÓM

STT MSSV Họ và Tên Chức Vụ Github Email
1 21522282 Lê Thị Liên Nhóm trưởng lltlien 21522282@gm.uit.edu.vn
2 21522883 Võ Thị Phương Anh Thành viên vtpanh 21522883@gm.uit.edu.vn

GIỚI THIỆU MÔN HỌC

  • Tên môn học: Nhập môn thị giác máy tính
  • Mã môn học: CS231
  • Mã lớp: CS231.N21.KHTN
  • Năm học: HK2 (2022 - 2023)
  • Giảng viên: TS.Mai Tiến Dũng

OVERVIEW

This repository contains the source code for an eyeglasses detection system developed as part of the CS231 Computer Vision course. The project focuses on implementing algorithms and techniques for detecting eyeglasses in images.

DEMO

Input Output
drawing drawing

Installation

git clone https://github.com/lltlien/eyeglasses-detection.git
cd eyeglasses-detection

Dependencies

deepface==0.0.68
opencv-python==4.5.5.62
numpy==1.24.3
imutils==0.5.4
tensorflow==2.0.1
scikit-learn==0.24.1
matplotlib==3.5.1

Training

  • To train the eyeglasses detection model using MobileNetV2.
python train_mobilenetv2_model.py
  • To train the eyeglasses detection model using ResNet101V2.
python train_resnetv2.py
  • Save trained model to predict.

Prediction

To detect all faces in an image and then predict eyeglasses using the trained models.

python image_prediction_using_deep_face.py

About

The final project for my class Computer Vision : Eyeglasses Detection using deepface, Mobilenetv2 and Resnet101v2


Languages

Language:Python 100.0%