Skip to content

This is an earmuff detector that uses video streams from CCTV cameras in workplaces to detect whether or not a person in the video is wearing earmuffs.

License

Notifications You must be signed in to change notification settings

BunNybuger/EarMuff_detector

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

43 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

EarMuff Detected EarMuff NOTDetected

EarMuff_detector

GunShop_Trim_OUTPUT.mp4

This repo contains an earmuff detector that uses video streams from CCTV cameras in workplaces to detect whether or not a person in the video is wearing earmuffs.

Open In Colab

We've used YoloV5 models to detect the person in the video and then their head, which is then sent to our PrototypicalNetwork for further analysis. We've trained our network with a small number of images (8 to be exact), and the results are promising. We've also included a sample video for you to check out. Due to the lack of a dedicated earmuff dataset, we employed a technique called Few Shot Learning (FSL) and used the efficientnet_b2 network architecture pre-trained on the ImageNet dataset as a feature extractor. We acknowledge that there is still room for improvement, and in future, we will be working on expanding our dataset and testing other methods to improve the accuracy. We'll keep this repository updated with our progress. And a big thanks to @ebennequin for his contribution in Few-Shot Learning.

So, what are you waiting for? Dive in, check out our work and let us know what you think!

About

This is an earmuff detector that uses video streams from CCTV cameras in workplaces to detect whether or not a person in the video is wearing earmuffs.

Topics

Resources

License

Stars

Watchers

Forks

Languages