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 prospective 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 masses data using
preprocessing.image_dataset_from_directory. You will achieve practical experience with the following concepts:
- Examine and understand data
- Build an enter pipeline
- Build the model
- Train the model
- Test the model
- Improve the model and repeat the process
Import TensorFlow and other libraries
Download and discover 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 wholesome grains image so here the model has predicted 100 % wholesome grains and as I have used object detection so the red mark have been showing few wholesome grains on the image.