The objective is to train a YOLO algorithm to detect multiple heads from a frame.
- TensorFlow
- Keras
Download full_yolo_backend.h5 from https://onedrive.live.com/?authkey=%21AM2OzK4S4RpT%2DSU&id=5FDEBAB7450CDD92%21121&cid=5FDEBAB7450CDD92
Put it in the root directory.
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
Run train.py
The weights for the front-end will be saved in the file name "model.h5".
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
This project is licensed under the MIT License