← 返回 AI 开源导航

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")

五、注意事项

  • 训练需要大量算力与数据;
  • 模型许可需逐一确认;
  • 推理时注意图像预处理一致性。