Veg_detect_ML_model

Machine Learning model trained to detect vegitables in Agro-industry

A Reliable Approach to Vegitable Detection in Agriculture with YOLOv5

🌟 Conclution :

• After thorough comparison of all metrics; training loss, validation loss, and mAP across all models, it is evident that all the models are performing satisfactorily.

• However, upon inspecting the predicted images, two models stood out for their exceptional detection accuracy and minimal false positives.

⚡ Model Training 2

Train parameters :  25 epochs, Adam, lr0 = 0.001, lrf = 0.001

[ Best.pt ] Github directory 👉 Veg_dtect/yolov5/runs/train/exp8/weights/best.pt

[ Best.pt ] Colab-Mydrive 👉 /content/drive/MyDrive/Veg_dtect/yolov5/runs/train/exp8/weights/best.pt

⚡ Model Training 3

Train parameters : 40 epochs, AdamW, lr0 = 0.0001, lrf = 0.0001

[ Best.pt ] path Github directory 👉 Veg_dtect/yolov5/runs/train/exp10/weights/best.pt

[ Best.pt ] path Mydrive 👉 /content/drive/MyDrive/Veg_dtect/yolov5/runs/train/exp10/weights/best.pt