google / aiyprojects-raspbian

API libraries, samples, and system images for AIY Projects (Voice Kit and Vision Kit)

Home Page:https://aiyprojects.withgoogle.com/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

[need help] Where is the list of possible classes for image classification

yangliu2 opened this issue · comments

video_capture.py said there is a file image_classification_classes.py that indicate the possible classes it will recognize. I did not find the file and not sure where is list of classes.
Could anybody point me to the right direction?
Appreciate it!

On the Pi, try looking at /opt/aiy/models/mobilenet_v1_160res_0.5_imagenet_labels.txt

(as per https://github.com/google/aiyprojects-raspbian/blob/aiyprojects/src/aiy/vision/models/image_classification.py#L38)

Good find. Laziness came over me I guess.