search

google / automl

Google Brain AutoML
Apache License 2.0
6.25k stars 1.45k forks source link

About ImageNet-21K Pretrain #1048

Closed SJLeo closed 3 years ago

SJLeo commented 3 years ago

The base tagging of the original imagenet21K is single label. I want to know how to get multi-label information for each image in ImageNet22K.

mingxingtan commented 3 years ago

The latest checkpoints are multiple, you can check out the tutorial to find out how to get the multi-label.

mingxingtan commented 3 years ago

I meant this tutorial: https://colab.sandbox.google.com/github/google/automl/blob/master/efficientnetv2/tfhub.ipynb

(set score threshold to 0.5)