| license: apache-2.0 | |
| pipeline_tag: image-feature-extraction | |
| ONNX port of [Unicom](https://arxiv.org/abs/2304.05884) model from [open-metric-learning](https://github.com/OML-Team/open-metric-learning). | |
| This model is intended to be used for similarity search. | |
| ### Usage | |
| Here's an example of performing inference using the model with [FastEmbed](https://github.com/qdrant/fastembed). | |
| ```py | |
| from fastembed import ImageEmbedding | |
| images = [ | |
| "./path/to/image1.jpg", | |
| "./path/to/image2.jpg", | |
| ] | |
| model = ImageEmbedding(model_name="Qdrant/Unicom-ViT-B-32") | |
| embeddings = list(model.embed(images)) | |
| # [ | |
| # array([ 0.04177791, 0.0550059 , 0.00025418, 0.0252876 , ..., dtype=float32), | |
| # array([2.23932182e-03, 4.68995124e-02, 3.28772422e-03, 7.57176951e-02, ...], dtype=float32) | |
| # ] | |
| ``` |