head-detection-using-yolo
head-detection-using-yolo copied to clipboard
pranoyr
Detection of head using YOLO
Head Detection Using YOLO Algorithm
The objective is to train a YOLO algorithm to detect multiple heads from a frame.
Getting Started
Prerequisites
- TensorFlow
- Keras
Download the pre-trained weights for the backend
Download full_yolo_backend.h5 from https://drive.google.com/file/d/1Q9WhhRlqQbA4jgBkCDrynvgquRXZA_f8/view?usp=sharing Put it in the root directory.
Dataset
Download the dataset and put it in the root directory.
Images - https://drive.google.com/open?id=1zn-AGmsBqVheFPnDTXWBpeo3XRH1Ho15
Annotations - https://drive.google.com/open?id=1LiTDMWk0KglGueJCaxgneEA_ltvEbUDV
Training
Run train.py
The weights for the front-end will be saved in the file name "model.h5".
Inference
I have uploaded the front-end weights. https://drive.google.com/file/d/1wg4q9cc6q04oRr_Xaf9GVhhbTSH4-ena/view?usp=sharing
Give the path to your image in predict.py
Run predict.py
License
This project is licensed under the MIT License
pranoyr