class_docstr: Text-query-based OWL-ViT model.
  
  This module computes joint text and image embeddings which are then
  used for localized prediction of bounding boxes and classes.
  
  Attributes:
    body_configs: Configurations of the image-text module.
    objectness_head_configs: Configurations for the (optional) objectness head.
    mask_head_configs: Configurations for the (optional) mask head.
    normalize: Whether to normalize the output of the model and the
      label_embeddings before computing the class logits.
    box_bias: Type of box bias - one of 'location', 'size' or 'both'.
class_docstr_tok: ["Text-query-based","OWL-ViT","model",".","This","module","computes","joint","text","and","image","embeddings","which","are","then","used","for","localized","prediction","of","bounding","boxes","and","classes",".","Attributes",":","body_configs",":","Configurations","of","the","image-text","module",".","objectness_head_configs",":","Configurations","for","the","(","optional",")","objectness","head",".","mask_head_configs",":","Configurations","for","the","(","optional",")","mask","head",".","normalize",":","Whether","to","normalize","the","output","of","the","model","and","the","label_embeddings","before","computing","the","class","logits",".","box_bias",":","Type","of","box","bias","-","one","of","'location","'",",","'size","'","or","'both","'","."]
docstr: Embeds images into feature maps.
  
  Args:
    images: images of shape (batch, input_size, input_size, 3), scaled to the
      input range defined in the config. Padding should be at the bottom right
      of the image.
    train: Whether or not we are in training mode.
  
  Returns:
    A 2D map of image features.
docstr_tok: ["Embeds","images","into","feature","maps",".","Args",":","images",":","images","of","shape","(","batch",",","input_size",",","input_size",",","3",")",",","scaled","to","the","input","range","defined","in","the","config",".","Padding","should","be","at","the","bottom","right","of","the","image",".","train",":","Whether","or","not","we","are","in","training","mode",".","Returns",":","A","2D","map","of","image","features","."]
code: def image_embedder(
      self, images: jnp.ndarray, train: bool
  ) -> jnp.ndarray:
      """Embeds images into feature maps.
  
      Args:
        images: images of shape (batch, input_size, input_size, 3), scaled to the
          input range defined in the config. Padding should be at the bottom right
          of the image.
        train: Whether or not we are in training mode.
  
      Returns:
        A 2D map of image features.
      """
      image_features, _ = self._embedder(images=images, train=train)
      return utils.seq2img(images, image_features)
  
imports: import copy
  from typing import Any, Dict, List, Mapping, Optional
  import flax.linen
  from flax.training import checkpoints
  import jax
  import jax.numpy
  import ml_collections
  from scenic.projects.owl_vit import layers
  from scenic.projects.owl_vit import matching_base_models
  from scenic.projects.owl_vit import utils
  from scenic.projects.owl_vit.clip import model
  from scenic.projects.owl_vit.clip import tokenizer
label_desc: Use image node_id 8 for calling the TextZeroShotDetectionModule obj's underlying member method code with example usage: obj.image_embedder(images, train) and returns: utils