Actual cats and dogs

We've demonstrated our new tools on the notMNIST dataset, which was helpful as it served to provide a comparison to our earlier simpler network setup. Now, let's progress to a more difficult problem—actual cats and dogs.

We'll utilize the CIFAR-10 dataset. There will be more than just cats and dogs, there are 10 classes—airplanes, automobiles, birds, cats, deer, dogs, frogs, horses, ships, and trucks. Unlike the notMNIST set, there are two major complexities, which are as follows:

  • There is far more heterogeneity in the photos, including background scenes
  • The photos are color

We have not worked with color datasets before. Luckily, it is not that different from the usual black and white dataset—we will just add another ...

Get Machine Learning with TensorFlow 1.x now with the O’Reilly learning platform.

O’Reilly members experience books, live events, courses curated by job role, and more from O’Reilly and nearly 200 top publishers.