Building an object detection model

We'll now make some code by using TFoS and Sparkdl. The dataset consists of images of buses and cars that have been curated from a Google image search. The aim is to train a model so that it can differentiate between cars and buses. The following is a list of prerequisites that you will need for this code to work:

  • PySpark
  • Python
  • TensorFlow
  • TensorFlowOnSpark
  • Pillow
  • Keras
  • TensorFrames
  • Wrapt
  • pandas
  • FindSpark
  • py4j

First, let's explore our dataset. Inception v3 was trained on ImageNet data with 1,000 categories. These included images of various vehicles as well. We have 49 images for buses and 41 images of cars. Here, you can see the sample images from the dataset:

Now, let's build the code:

  1. This time, we'll ...

Get TensorFlow Machine Learning Projects 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.