EVA-CLIP: Improved Training Techniques for CLIP at Scale
Paper • 2303.15389 • Published
How to use birder-project/rope_i_vit_l14_nf_swiglu_c1_eva02-clip with Birder:
# No code snippets available yet for this library. # To use this model, check the repository files and the library's documentation. # Want to help? PRs adding snippets are welcome at: # https://github.com/huggingface/huggingface.js
A RoPE ViT-L14 image encoder from the EVA02 CLIP model by Sun et al., converted to the Birder format for image feature extraction. This version retains the original model weights and architecture. It is a general-purpose visual backbone.
See: https://huggingface.co/QuanSun/EVA-CLIP for further details.
Model Type: Image classification and detection backbone
Model Stats:
Papers:
import birder
from birder.inference.classification import infer_image
# Option 1: manual setup (more control over preprocessing)
net, model_info = birder.load_pretrained_model("rope_i_vit_l14_nf_swiglu_c1_eva02-clip", inference=True)
# Get the image size the model was trained on
size = birder.get_size_from_signature(model_info.signature)
# Create an inference transform
transform = birder.classification_transform(size, model_info.rgb_stats)
# Option 2: helper (quick start with default preprocessing)
net, model_info, transform = birder.load_pretrained_model_and_transform("rope_i_vit_l14_nf_swiglu_c1_eva02-clip", inference=True)
image = "path/to/image.jpeg" # or a PIL image, must be loaded in RGB format
out, _ = infer_image(net, image, transform)
# out is a NumPy array with shape of (1, 768), representing class probabilities.
import birder
from birder.inference.classification import infer_image
# Option 1: manual setup (more control over preprocessing)
net, model_info = birder.load_pretrained_model("rope_i_vit_l14_nf_swiglu_c1_eva02-clip", inference=True)
# Get the image size the model was trained on
size = birder.get_size_from_signature(model_info.signature)
# Create an inference transform
transform = birder.classification_transform(size, model_info.rgb_stats)
# Option 2: helper (quick start with default preprocessing)
net, model_info, transform = birder.load_pretrained_model_and_transform("rope_i_vit_l14_nf_swiglu_c1_eva02-clip", inference=True)
image = "path/to/image.jpeg" # or a PIL image
out, embedding = infer_image(net, image, transform, return_embedding=True)
# embedding is a NumPy array with shape of (1, 1024)
from PIL import Image
import birder
net, model_info, transform = birder.load_pretrained_model_and_transform("rope_i_vit_l14_nf_swiglu_c1_eva02-clip", inference=True)
image = Image.open("path/to/image.jpeg")
features = net.detection_features(transform(image).unsqueeze(0))
# features is a dict (stage name -> torch.Tensor)
print([(k, v.size()) for k, v in features.items()])
# Output example:
# [('stage1', torch.Size([1, 1024, 24, 24]))]
@article{Fang_2024,
title={EVA-02: A visual representation for neon genesis},
volume={149},
ISSN={0262-8856},
url={http://dx.doi.org/10.1016/j.imavis.2024.105171},
DOI={10.1016/j.imavis.2024.105171},
journal={Image and Vision Computing},
publisher={Elsevier BV},
author={Fang, Yuxin and Sun, Quan and Wang, Xinggang and Huang, Tiejun and Wang, Xinlong and Cao, Yue},
year={2024},
month=Sept, pages={105171}
}
@misc{sun2023evaclipimprovedtrainingtechniques,
title={EVA-CLIP: Improved Training Techniques for CLIP at Scale},
author={Quan Sun and Yuxin Fang and Ledell Wu and Xinlong Wang and Yue Cao},
year={2023},
eprint={2303.15389},
archivePrefix={arXiv},
primaryClass={cs.CV},
url={https://arxiv.org/abs/2303.15389},
}
Base model
QuanSun/EVA-CLIP