ViT Feature Extractor¶
Overview¶
This page includes information about how to use ViTFeatureExtractorTF with tensorflow-ops.
This feature extractors works in sync with Dataset and so is useful for on the fly preprocessing.
>>> from tf_transformers.models import ViTFeatureExtractorTF
>>> image_path_list = # List fo image paths
>>> vit_feature_extractor_tf = ViTFeatureExtractorTF(img_height=224, img_width=224)
>>> outputs = vit_feature_extractor_tf({'image': tf.constant(image_path_list)})