Openclip
一、项目简介
OpenCLIP 是开源的 CLIP 模型复现与改进实现,提供多种规模的图文对比学习模型与完整训练代码。相比原始 CLIP,它更开放、可复现,方便研究者在自己的数据集上训练图文向量模型。
二、核心特性
- 开源可复现:完整训练与评测代码;
- 多规模模型:从小型到大型;
- 自定义数据训练:支持自有图文数据;
- 与 HF 生态兼容;
- 社区持续更新;
- 预训练权重丰富。
三、适用场景
- 图文检索与匹配;
- 零样本分类;
- 多模态向量化研究。
四、快速上手
pip install open-clip-torch
import open_clip
model, _, preprocess = open_clip.create_model_and_transforms("ViT-B-32", pretrained="laion2b_s34b_b79k")
tok = open_clip.get_tokenizer("ViT-B-32")
五、注意事项
- 训练需要大量算力与数据;
- 模型许可需逐一确认;
- 推理时注意图像预处理一致性。