https://github.com/pranoyr/head-detection-using-yolo

Detection of head using YOLO
https://github.com/pranoyr/head-detection-using-yolo

Last synced: 2 months ago
JSON representation

Detection of head using YOLO

• Host: GitHub
• URL: https://github.com/pranoyr/head-detection-using-yolo
• Owner: pranoyr
• License: mit
• Created: 2018-06-23T18:55:15.000Z (over 6 years ago)
• Default Branch: master
• Last Pushed: 2023-03-24T22:24:24.000Z (almost 2 years ago)
• Last Synced: 2024-02-11T22:47:24.709Z (11 months ago)
• Language: Python
• Size: 1.8 MB
• Stars: 158
• Watchers: 6
• Forks: 53
• Open Issues: 12
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

• awesome-face-detection-and-recognition - pranoyr/head-detection-using-yolo - detection-using-yolo?style=social"/> : Detection of head using YOLO. (Face Detection)
• awesome-yolo-object-detection - pranoyr/head-detection-using-yolo - detection-using-yolo?style=social"/> : Detection of head using YOLO. (Applications)
• awesome-yolo-object-detection - pranoyr/head-detection-using-yolo - detection-using-yolo?style=social"/> : Detection of head using YOLO. (Applications)

README

        
# Head Detection Using YOLO Algorithm

The objective is to train a YOLO algorithm to detect multiple heads from a frame.

## Getting Started

### Prerequisites

1. TensorFlow

2. 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