/coffee-detection

👨🏽‍💻 College graduation project aimed at creating an object detection custom model in TensorFlow that performs recognition and counting of coffee plants in a coffee plantation.

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

Machine learning model to detection and counting of coffee feet (Coffea sp.) by video analysis

Advisor: João Ricardo Favan
Authors: Andrew Fernandes and Felipe Jonas

ABSTRACT

The development of the research was based on the following stages for its realization: preparation of the images, configuration of the pre-trained models for customized training to the coffee trees, training of the model, testing of model with images captured by the authors at the “Shuji Nishimura” Technology College in the city of Pompeia, implementation of the model for detection in coffees trees in video format and development of the cumulative count of tree. The result generated was a machine learning model trained to detect coffee trees by video and cumulatively count each unit in the planting row. However, the adoption of this type of technology can contribute positively to the analysis of productivity by planted area and work together with specialists and companies that want to have this technology in their equipment.

key-words

Object detection; Machine learning; Plant count.

Minimal Requirements

Python | PyPI

Dependencies

pip install -r requirements.txt

Usage

python tensorflow_cumulative_object_counting.py --model=Model Path --labelmap=Path to Labelmap --video_path=Path to video
python tensorflow_cumulative_object_counting.py --model=inference_graph/saved_model --labelmap=coffee_counter_training/training/labelmap.pbtxt --video_path=input_videos/coffee_plantation.mp4

How to contribute

  • Make a fork of this repository
  • Clone to you machine and entry on respective paste
  • Create a branch with your resource: git checkout -b my-feature
  • Commit your changes: git commit -m 'feat: My new feature'
  • Push your branch: git push origin my-feature

Find and report issues
Star the project

License

GNU General Public License
This work is licensed under a version 3 of the GNU General Public License.