Image Classification using GluonCV

Implement a tennis ball detector using a pre-trained image classification network from GluonCV

Ramesh Paudel, Ph.D.
Towards Data Science
4 min readMay 25, 2020

--

In the previous article, we discussed about object detection using GluonCV. In this article, we will discuss how to implement a binary image classifier that classify whether a given image is a tennis ball or not using a pre-trained image classification network from GluonCV. We implement the machine learning pipeline step by step, from loading and transforming an input image, to loading and using a pre-trained model.

--

--