Mastering Computer Vision with TensorFlow 2.x

Image classification

Now let's perform the following steps for image classification:

1. Go to the image_classification directory, that is, pi/examples/lite/examples/image_classification/raspberry_pi, using the File Manager. You will see a file named classify_picamera.py. Now, go to https://www.tensorflow.org/lite/guide/hosted_models and download the object detection models and label files folder named mobilenet_v2_1.0_224.tflite and the labels_mobilenet_v2_1.0_224.txt file. Copy those files into pi/examples/lite/examples/image_classification/raspberry_pi.

2. Next, go to the directory using the Terminal using pi/examples/lite/examples/image_classification/raspberry_pi and execute the following command:

$Python3 classify_picamera.py –model mobilenet_v2_1.0_224.tflite ...