ImageEncoder

oxirs_vec::cross_modal_embeddings

Trait ImageEncoder 

Source 
pub trait ImageEncoder: Send + Sync {
    // Required methods
    fn encode(&self, image: &ImageData) -> Result<Vector>;
    fn encode_batch(&self, images: &[ImageData]) -> Result<Vec<Vector>>;
    fn get_embedding_dim(&self) -> usize;
    fn extract_features(&self, image: &ImageData) -> Result<Vec<f32>>;
}
Expand description

Image encoder trait for cross-modal systems

Required Methods§

Source

fn encode(&self, image: &ImageData) -> Result<Vector>

Source

fn encode_batch(&self, images: &[ImageData]) -> Result<Vec<Vector>>

Source

fn get_embedding_dim(&self) -> usize

Source

fn extract_features(&self, image: &ImageData) -> Result<Vec<f32>>

Implementors§