View Source Bumblebee.Vision.DeitFeaturizer (Bumblebee v0.4.2)

DeiT featurizer for image data.

Configuration

:resize - whether to resize (and optionally center crop) the input to the given :size. Defaults to true
:size - the size to resize the input to. Either a single number or a {height, width} tuple. Only has an effect if :resize is true . Defaults to 256
:resize_method - the resizing method, either of :nearest, :bilinear, :bicubic, :lanczos3, :lanczos5. Defaults to :bicubic
:center_crop - whether to crop the input at the center. If the input size is smaller than :crop_size along any edge, the image is padded with zeros and then center cropped . Defaults to true
:crop_size - the size to center crop the image to. Only has an effect if :center_crop is true. Defaults to 224
: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.485, 0.456, 0.406]
:image_std - the sequence of standard deviations for each channel, to be used when normalizing images. Defaults to [0.229, 0.224, 0.225]