Salisbury-University / robot-weed-killer

Application to control our laser firing, weed killing robot
5 stars 1 forks source link

💡 Train Weed Detection Model on Weed Dataset #48

Closed qiugai closed 1 year ago

qiugai commented 1 year ago

Start Date

3/7/23

Implementation PR

No response

Reference Issues

No response

Summary

We need to change from our generic yolov5 general object dataset, to a custom model using a dataset of weeds on sidewalks and pavements.

Basic Example

Detect weeds not fridges and people and chairs

Drawbacks

Training take long time :(

qiugai commented 1 year ago

Data is being trained on Roboflow, ~1000 images, 50% | 30% | 20% | Train | Valid | Test split

ZM009 commented 1 year ago

Experienced the results of the training yesterday. Holy moly

ZM009 commented 1 year ago

Data training worked fine, model had some different results when training more.