pub struct CLIPModel { /* private fields */ }Expand description
CLIP model for image-text embeddings
Implementations§
Source§impl CLIPModel
impl CLIPModel
Sourcepub fn new(model_path: &str) -> RragResult<Self>
pub fn new(model_path: &str) -> RragResult<Self>
Create new CLIP model
Sourcepub fn generate_embedding(
&self,
_image: &PreprocessedImage,
) -> RragResult<Vec<f32>>
pub fn generate_embedding( &self, _image: &PreprocessedImage, ) -> RragResult<Vec<f32>>
Generate CLIP embedding for image
Sourcepub fn generate_text_embedding(&self, _text: &str) -> RragResult<Vec<f32>>
pub fn generate_text_embedding(&self, _text: &str) -> RragResult<Vec<f32>>
Generate text embedding for comparison
Sourcepub fn calculate_similarity(
&self,
image: &PreprocessedImage,
text: &str,
) -> RragResult<f32>
pub fn calculate_similarity( &self, image: &PreprocessedImage, text: &str, ) -> RragResult<f32>
Calculate similarity between image and text
Auto Trait Implementations§
impl Freeze for CLIPModel
impl RefUnwindSafe for CLIPModel
impl Send for CLIPModel
impl Sync for CLIPModel
impl Unpin for CLIPModel
impl UnwindSafe for CLIPModel
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