← Machine Learning & AI

04Machine Learning & AI

Trained Real-Time Face Mask Detection System with YOLOv4

Done

What I built

Trained and evaluated a YOLOv4 deep learning object detection pipeline in Python using OpenCV and Darknet/PyTorch to recognize face mask compliance in real-time camera video streams with high frame rates.

Implemented custom dataset preprocessing, bounding box annotation parsing, and inference pipelines for multi-face detection.

What clicked

Dataset augmentation (rotation, illumination shifts, occlusion) and careful anchor box tuning are crucial for computer vision models to generalize across varied real-world camera angles and lighting environments.

What I'd change

Quantize the model using ONNX / TensorRT to lower latency for real-time edge deployment on cost-effective embedded hardware.