pub struct ImageEncoder { /* private fields */ }Expand description
ViT image encoder producing a dense image embedding.
Implementations§
Source§impl ImageEncoder
impl ImageEncoder
Sourcepub fn forward(&self, image: &[f32]) -> VisionResult<FeatureMap>
pub fn forward(&self, image: &[f32]) -> VisionResult<FeatureMap>
Encode a flat CHW image into a dense (embed_dim, grid, grid) map.
§Errors
Propagates patch-embed / block errors.
Auto Trait Implementations§
impl Freeze for ImageEncoder
impl RefUnwindSafe for ImageEncoder
impl Send for ImageEncoder
impl Sync for ImageEncoder
impl Unpin for ImageEncoder
impl UnsafeUnpin for ImageEncoder
impl UnwindSafe for ImageEncoder
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more