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

google / aiyprojects-raspbian

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

https://aiyprojects.withgoogle.com/

Apache License 2.0

1.62k stars 694 forks source link

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

Closed yangliu2 closed 3 years ago

yangliu2 commented 3 years ago

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!

scottamain commented 3 years ago

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)

yangliu2 commented 3 years ago

Good find. Laziness came over me I guess.