transformers pytorch tf data2vec-vision image-feature-extraction image-classification vision dataset:imagenet dataset:imagenet-1k arxiv:2202.03555 arxiv:2106.08254 license:apache-2.0 endpoints_compatible region:us