class_docstr: Wraps an OWL-ViT FLAX model for convenient inference.
  
  All public methods apply to a single example and take and return Numpy arrays.
  
  Attributes:
    config: ConfigDict with model configuration.
    module: OWL-ViT Flax module.
    variables: Variable dict to be used with module.apply.
docstr: Embeds image and returns image embeddings and boxes.
  
  Args:
    image: Single uint8 Numpy image of any size. Will be converted to float
      and resized before passing it to the model.
  
  Returns:
    Numpy arrays containing image features, class embeddings, and predicted
    boxes.
docstr_tok: ["Embeds","image","and","returns","image","embeddings","and","boxes",".","Args",":","image",":","Single","uint8","Numpy","image","of","any","size",".","Will","be","converted","to","float","and","resized","before","passing","it","to","the","model",".","Returns",":","Numpy","arrays","containing","image","features",",","class","embeddings",",","and","predicted","boxes","."]
code: def embed_image(
      self, image: np.ndarray
  ) -> Tuple[np.ndarray, np.ndarray, np.ndarray]:
      """Embeds image and returns image embeddings and boxes.
  
      Args:
        image: Single uint8 Numpy image of any size. Will be converted to float
          and resized before passing it to the model.
  
      Returns:
        Numpy arrays containing image features, class embeddings, and predicted
        boxes.
      """
      image = self.preprocess_image(image)
      out = self._embed_image_jitted(image[None, ...])
      return jax.tree_util.tree_map(lambda x: np.array(x[0]), out)
  
imports: import dataclasses
  import functools
  from typing import Any, Dict, Tuple
  from flax import linen
  import jax
  import jax.numpy
  import ml_collections
  import numpy
  from scenic.model_lib.base_models import box_utils
  from scenic.projects.owl_vit.notebooks import numpy_cache
  from scipy import special
  from skimage import transform
  import tensorflow
label_desc: Use image node_id 5 for calling the Model obj's underlying member method code with example usage: obj.embed_image(image) and returns: jax