/Wheat-detection

Wheat heads detection using Faster-RCNN in pytorch

Primary LanguageJupyter Notebook

Wheat detection using faster RCNN in pytorch

The notebook describes how to train a Faster RCNN network to detect wheat heads from outdoor images of wheat plants.
Data can be downloaded from the kaggle competition page. The model was trained on google colab.

alt text