Skip to content

Latest commit

 

History

History
18 lines (14 loc) · 772 Bytes

readme.md

File metadata and controls

18 lines (14 loc) · 772 Bytes

Gender Identifier YOLOv5

Identify if someone is a male or female based off of their face.

Model

Trained with the YOLOv5 computer vision library. Has over 90% accuracy(see confusion matrix below).
Confusion Matrix

Dataset

I scraped and annotated my own, custom dataset. Contains over 2.6k images.
Dataset can be found here: https://universe.roboflow.com/generalstuff/women-men/dataset/1

Webcam Integration

I followed a tutorial to add webcam integration with Flask. I added the YOLOv5 integration myself.
Webcam Integration Tutorial: https://blog.miguelgrinberg.com/post/video-streaming-with-flask

Demo

The demonstration video can be found here: Coming Soon
A less laggy version can be found here: Coming Soon