Identifying weed plants in cultivated filed is a time-intensive task. Automating such a task can drastically reduce the time needed to identify and remove weeds and thus increase the yield and the productivity of the workers. In our latest project, we worked on a plant segmentation task that uses Deep Learning to identify weeds and their stems from crops.
In this repositry, we share with you our implementations and pretrained weights for the following models:
- U-Net model with 4-channels
- ResNet model with 4-channels
- U-Net Joint Model
Please refer to the following article where we document our journey starting from finding relevant datasets, choosing the proper neural network architecture, training the model, evaluating its performance, and finally launching it.