Classifying images with MXNet – GluonCV Model Zoo, AlexNet, and ResNet

MXNet provides a variety of tools to compose custom deep learning models. In this recipe, we will see how to use MXNet to build a model from scratch, train it, and use it to classify images from a dataset. We will also see that although this approach works fine, it is time-consuming.

Another option, and one of the highest value features that MXNet and GluonCV provide, is their Model Zoo. GluonCV Model Zoo is a set of pre-trained, ready-to-go models, for use with your own applications. We will see how to use Model Zoo with two very important models for image classification – AlexNet and ResNet.

In this recipe, we will analyze and compare these approaches to classify images on a reduced version of the Dogs vs. Cats dataset.

Getting ready

As with previous chapters, in this recipe, we will use a few matrix operations and linear algebra, but it will not be too difficult.

Furthermore, we will...