View Source Bumblebee.Vision.VitFeaturizer (Bumblebee v0.6.0)

ViT featurizer for image data.

Configuration

• :resize - whether to resize the input to the given :size. Defaults to true

• :size - the size to resize the input to, given as %{height: ..., width: ...}. Only has an effect if :resize is true . Defaults to %{width: 224, height: 224}

• :resize_method - the resizing method, either of :nearest, :bilinear, :bicubic, :lanczos3, :lanczos5. Defaults to :bilinear

• :normalize - whether or not to normalize the input with mean and standard deviation. Defaults to true

• :image_mean - the sequence of mean values for each channel, to be used when normalizing images. Defaults to [0.5, 0.5, 0.5]

• :image_std - the sequence of standard deviations for each channel, to be used when normalizing images. Defaults to [0.5, 0.5, 0.5]