google/aiyprojects-raspbian

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

yangliu2 opened this issue · 2 comments

yangliu2 commented

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

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

Good find. Laziness came over me I guess.