Skip to content

Disease classifier in strawberries, tomatoes, and beans using Computer Vision with TensorFlow and Keras. πŸƒπŸŒΏ

License

Notifications You must be signed in to change notification settings

JavierdiazS/Fruit-Infection-Disease

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

6 Commits
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

Fruit Infection Disease Classification

end

Final result:

gif

Summary

Classifying diseases in fruit-bearing plants such as strawberries and tomatoes, and in legumes such as beans, enables the precise identification of causative agents, such as bacteria, fungi, or viruses. This is essential for developing specific control and management strategies. Each pathogenic agent may require different prevention and treatment methods.

Accurate classification is crucial for international trade of agricultural products, and precise disease identification helps prevent the introduction of pathogens into new regions.

With the assistance of Deep Learning, we can automate this process.

This project was carried out using the crop disease dataset collected by Nikit Kashyap (2022) on the Kaggle platform. The dataset comprises 5500 images: 1166 for testing, 2907 for training, and 1427 for validation. This diverse collection of images represents various types of diseases affecting crops like strawberries, tomatoes, and beans here.

To conduct tests on the local Web server (testing the index.html), a dataset called "data_image" was created on Kaggle, containing 3 images created by me here.

Six different neural architectures were tested, yielding varied results, with the best performance achieved by the fourth and final neural architecture:

  • 1st neural network: loss: 2.7913 - accuracy: 0.7770
  • 2nd neural network: loss: 4.8144 - accuracy: 0.6732
  • 3rd neural network: loss: 3.5195 - accuracy: 0.5763
  • 4th neural network: loss: 1.6255 - accuracy: 0.7890 βœ…
  • 5th neural network: loss: 4.2998 - accuracy: 0.6604
  • 6th neural network (Inception_v3): loss: 1.1829 - accuracy: 0.8139 βœ…

Repo structure

The project has the following structure:

  • notebook/: .ipynb notebook for data eploration, training models and ensembling.
  • webapp/: Contains the .html and the command to run the web from the local server. Along with the .json of the latest model.

Contributing

Feel free to use it if you want to contribute directly to the code base.

License

It is released under the MIT license. See the LICENSE file for details.

About

Disease classifier in strawberries, tomatoes, and beans using Computer Vision with TensorFlow and Keras. πŸƒπŸŒΏ

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published