PPE-Detection-YOLO

The following are the steps you can follow to train your dataset in google colaboratory. The following are the steps you should follow to train your custom Yolo model:

1. Create dataset compatible with the YOLO format. LabelImg
2. Divide the dataset into train-test format.(train_test.py)
3. Modify the cfg file.
4. Prepare the data into a zip according to the google colaboratory.
5. Load the prepared data into the drive account of google.
6. Train the model at Google Colab
7. Save model in specific steps(1000-2000..).Here are the weight results of my dataset
8. Download model to test at local.
9. Finally, test the model for image, video and webcam.My video and webcam results