What is TensorFlow?
Tensorflow is an open-source library for numerical computation and large-scale machine learning that ease Google Brain TensorFlow, the process of acquiring data, training models, serving predictions, and refining future results.
How Tensorflow and Keras used for Image Classification
This blogs shows how to classify images of wheat classes. It creates an image classifier using a
keras.Sequential model, and loads data using
preprocessing.image_dataset_from_directory. You will gain practical experience with the following concepts:
- Examine and understand data
- Build an input pipeline
- Build the model
- Train the model
- Test the model
- Improve the model and repeat the process
Import TensorFlow and other libraries
Download and explore the dataset
Configure the Model for Training
After Training the model we get the model file in .h5 filename. and by using that model file we can generate the probability score of grains in the image. As of now we have taken a healthy grains image so here the model has predicted 100 % healthy grains and as I have used object detection so the red mark have been showing few healthy grains on the image.