With applications ranging from classifying objects in self driving cars to identifying blood cells in healthcare industry to identifying defective items in manufacturing industry, image classification is one of the most important applications of computer vision.
How does it work? Which framework should you use?
Here’s a great tutorial.
In this article, we will understand how to build a basic image classification model in PyTorch and TensorFlow. We will start with a brief overview of both PyTorch and TensorFlow. And then we will take the benchmark MNIST handwritten digit classification dataset and build an image classification model using CNN (Convolutional Neural Network) in PyTorch and TensorFlow.