Classes in the pretrained models in zoo

General
1

Hi there!
I would like to use some pretrained and already compiled models from the zoo on Hailo-8 by downloading the HEF file and using it. But how can I tell what each model was trained for? Under classification there are many models, but no information on what classes they will classify.
the information in the “source” link is very general and does not provide enough details.

Thanks!

2

Hi @srubin, each pre-trained model, was trained on a dataset, e.g. ImageNet or COCO. These dictates what are the classes.

3

@srubin As @Nadav mentioned, the classes are dependent on the dataset that the model was originally trained on.

You can find the dataset used for training in the model-zoo configuration files (.yaml files). For example, for yolov5m, you can see that the dataset used for training is coco 2017: