Object Detection, segmentation & classification using YOLO v8
https://github.com/ultralytics/ultralytics
- To train model on custom data set first annotate the test, train and validation model and generate ymal file
- Upload the data into Google drive as we will train the model over there into google colab using GPU
- Post traning best & last model will be saved into google drive
- Use best model for predection
https://github.com/HumanSignal/labelImg
Update the Yaml file with actual images(train, val, number of class & class list) file path of your google drive, Yolov8 was the folder where I have uploaded the train, val, test images
- train: /content/drive/MyDrive/Yolov8/data/train/images
- val: /content/drive/MyDrive/Yolov8/data/valid/images
- nc: 5
- names: ['Helmet', 'Goggles', 'Jacket', 'Gloves', 'Footwear']