10 分钟跑通 open_clip:不写一行标注的零样本多模态检索

发布时间:2026/11/13 1:31:14

10 分钟跑通 open_clip:不写一行标注的零样本多模态检索 10 分钟跑通 open_clip不写一行标注的零样本多模态检索【免费下载链接】open_clipAn open source implementation of CLIP.项目地址: https://gitcode.com/GitHub_Trending/op/open_clip搜索框里只敢填标签词你有 5 万张商品图搜索框里却只能填标签标一个新类目要几天类目一变又要重来。open_clip 把图像和文本编码进同一向量空间零样本检索开箱即用。最小可行上手路径一条命令装好依赖依赖版本备注Python 3.9pyproject.toml声明的最低版本torch 2.6低于此版本装不上torchvision与 torch 匹配图像预处理依赖timm 1.0.29ConvNeXt、SigLIP 等图像塔依赖pip install open_clip_torch # 后续要自己训练 CLIP 时再装 training 依赖组 pip install open_clip_torch[training]import torch from PIL import Image import open_clip model, _, preprocess open_clip.create_model_and_transforms( ViT-B-32, pretrainedlaion2b_s34b_b79k) model.eval() # 模型默认训练模式推理前必须切 eval tokenizer open_clip.get_tokenizer(ViT-B-32) image preprocess(Image.open(docs/CLIP.png)).unsqueeze(0) text tokenizer([a diagram, a dog, a cat]) with torch.no_grad(): image_features model.encode_image(image) text_features model.encode_text(text) # 省略两侧特征归一化算余弦相似度softmax 得标签概率权重首次运行自动下载之后离线可用。整个流程不到 10 分钟瓶颈只在首次下载权重。示例里对docs/CLIP.png打三个候选标签a diagram概率为 1.0README 官方示例。核心能力速查ViT-B-32 还是 ViT-H-14 怎么选 参数量取自docs/model_profile.csv零样本精度取自 README 的官方 ImageNet-1k 基准。表里参数量是图像塔加文本塔的总和。同一模型不同预训练权重精度不同选权重往往比选架构更影响效果。模型参数量ImageNet 零样本 Top-1适用场景推荐硬件ViT-B-32151M72.8%DataComp-1B 权重快速验证、显存紧张8GB 显存单卡ViT-B-16150M73.5%DataComp-1B 权重精度与吞吐平衡12GB 显存单卡ViT-L-14428M79.2%DataComp-1B 权重生产级跨模态检索24GB 显存单卡ViT-H-14986M78.0%LAION-2B 权重高精度离线任务多卡ViT-bigG-142540M80.1%LAION-2B 权重精度上限、集群训练多卡 A100不知道选哪个就选ViT-B-32laion2b_s34b_b79k。这是 README 示例的默认组合8GB 显存单卡就能跑。想换模型时用open_clip.list_pretrained()查可用权重。实战场景检索和训练两条最短路径场景一商品图以文搜图图库没有标签体系时把自然语言直接当查询词。标签换一句描述就能扩展新类目不用改任何代码。image_features model.encode_image(product_images) # 商品图批量编码 text_features model.encode_text(tokenizer(红色皮质运动鞋)) # 省略两侧归一化后算余弦相似度按得分取 top-k参考官方 38 数据集零样本评测docs/openclip_results.csvViT-B-16的 ImageNet 零样本 top-1 为 73.5%。场景二自建图文对从零训练数据整理成 webdataset每个样本同名.jpg加.txt后直接起训练。不想跑 ImageNet 零样本评估时删掉--imagenet-val即可省一份数据。python -m open_clip_train.main \ --model ViT-B-32 \ --train-data /data/train-{0000..2175}.tar \ --imagenet-val /data/imagenet/validation/ \ --batch-size128 --lr1e-3 --epochs30官方用 128 张 A100 在 LAION-400M 上训ViT-B/32耗时约 36 小时。ImageNet 零样本 top-1 为 62.96%出处docs/PRETRAINED.md。调优与踩坑清单六个高频问题 以下问题都可以用一行命令或参数修复。老权重加载后精度无故掉点→ 模型名换-quickgelu后缀如ViT-B-32-quickgelu→ 精度对齐官方基准ConvNeXt、SigLIP 报Unknown model→pip install -U timm→ 图像塔由 timm 提供需较新版本多卡训练显存被 logit 矩阵吃爆→ 加--local-loss --gather-with-grad→ 空间复杂度降到近似线性单卡 batch 上不去→ 先开--grad-checkpointing再用--accum-freq k等效放大 batch训练吞吐低、指标盯不过来→--torchcompile配--torchcompile-strategy task再用--report-to tensorboard盯 zero-shot 曲线线上推理想省显存→open_clip.utils.replace_linear换 bitsandbytesLinear8bitLt→ 官方教程实测量化层权重显存约降 2 倍CIFAR-10 零样本 int8 88.83% 对 fp16 88.76%延伸资源预训练模型列表38 数据集零样本基准结果模型参数量与 FLOPs 表训练脚本示例交互 NotebookInt8 推理教程【免费下载链接】open_clipAn open source implementation of CLIP.项目地址: https://gitcode.com/GitHub_Trending/op/open_clip创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
延伸阅读

更多相关文章

2026/11/12 2:14:29

SpringBoot+Netty实现物联网高并发通信方案

/* MD / 富文本中的 .toc(含博客园搬家等嵌套结构);.toc-box 在侧栏,不受影响 */#content_views .toc,/* 编辑器常在目录前后插入空 p(:empty 仍占 20px),一并去掉避免顶空隙 */#content_views.markdown_views > p:empty:has(+ .toc),#content_views.markdown_views …

2026/11/9 15:35:15

C++调用Python中文乱码根因与UTF-8闭环解决方案

1. 项目概述:为什么C调用Python时中文总像“天书”? 你写好了一段漂亮的C程序,用PyBind11或Python C API封装了核心算法,再用Python脚本调用它——结果一输出中文,控制台里全是问号、方块、小方格,甚至直接…

2026/11/13 10:18:19

隆阳区电工证去哪里考多少钱,实操考试心里没底怕挂科?

隆阳区电工证去哪里考多少钱,实操考试心里没底怕挂科? 实操考试心里没底怕挂科?这几乎是每个想考电工证的人最真实的写照。尤其是隆阳区电工证去哪里考这个问题,直接关系到你能不能顺利拿到证、能不能顺利上岗、甚至能不能拿到理想的薪资。别急,这篇文章…

2026/11/13 10:18:19

郑州人社电工证全国通用,实操考试心里没底怕挂科?这几点帮你稳过

郑州人社电工证全国通用,实操考试心里没底怕挂科?这几点帮你稳过 实操考试心里没底怕挂科?这是很多报考郑州人社电工证的小伙伴最真实的感受。尤其是对于刚入行的新人来说,面对高压电工证、低压电工证等实操项目,一不小心就可能挂科。但别担心,郑州人社…

2026/11/13 10:18:19

济宁电工证要怎么考?内部消息告诉你工地人怎么破局

济宁电工证要怎么考?内部消息告诉你工地人怎么破局 工地太忙根本没时间复习考试,尤其是那些在一线操作电工设备的工友,每天穿梭在钢筋水泥之间,连吃饭时间都紧张。可偏偏现在政策收紧,电工证不仅关系到能不能上岗,还直接影响到工资和职业发展。如果你在…

2026/11/13 10:18:19

办理慈溪电工证哪里考需要什么材料

你去哪报名慈溪电工证?真实经历告诉你别被中介坑 你是不是也和我一样,拿着手机在搜索“慈溪电工证哪里考”,结果一打开网页,全是各种培训机构、中介公司,打着“包过”“快速拿证”的旗号,搞得你心里直打鼓?我之前也是这样,生怕被坑了钱还拿不到证,花…

2026/11/13 10:13:19

特种工退休补助官方回应:工地太忙根本没时间复习考试怎么办?

特种工退休补助官方回应:工地太忙根本没时间复习考试怎么办? 工地太忙根本没时间复习考试,这是很多特种工朋友的共同困扰。尤其是涉及【特种工退休补助】的证书,很多人并不清楚它的实际价值,更别提如何准备考试了。官方回应明确表示,特种工在退休后确实…

2026/11/12 20:35:13

电工证丢失还能补吗?千万别踩坑,丢了证等于失业

电工证丢失还能补吗?千万别踩坑,丢了证等于失业 你正在找工作,但手里的电工证突然丢了,这下怎么办?别急,先别慌, 电工证丢失还能补吗 ?答案是肯定的,但 千万别踩坑 ,补办流程如果不了解,可能会耽误你上岗、甚至影响到你的收入。…

2026/11/13 0:01:06

办上海高压电工证良心建议:别被中介坑了,自己看这篇就够了

办上海高压电工证良心建议:别被中介坑了,自己看这篇就够了 你是不是也像我一样,第一次听说“高压电工证”这个名词,就感觉一头雾水?其实很多人都跟我一样,不知道去哪报名、怕被中介坑,结果花了冤枉钱还考不上。今天我就从一个在商丘培训过的学员角度,…

2026/11/13 0:01:06

黎里考电工证在哪里值不值得考

黎里考电工证在哪里值不值得考 之前考的证在外省能不能转过来?这是很多在外地工作、准备回老家发展的朋友最关心的问题。特别是在黎里地区,很多从事电力相关工作的朋友,都面临这样一个难题: 外省的电工证能否在本地使用?值不值得考?…

2026/11/13 0:01:06

毕节哪里考电工证 怎么报名学历不高也能上手

毕节哪里考电工证 怎么报名学历不高也能上手 你是不是也和我一样,学历不高,想着考个电工证,但又担心自己条件不够,怕报不上名?其实不然,只要符合条件,学历不高也能考电工证。今天就来聊聊【毕节哪里考电工证】这个问题,还有怎么报名,让你一步到位不…

2026/11/12 22:10:13

面向复杂光照与角度的宠物面部识别鲁棒性增强策略

在智慧养宠时代,基于视觉的宠物身份识别技术正逐步从实验室走向家庭应用。然而,与配合度高的人类不同,宠物是典型的“非合作生物识别对象”——它们不会主动配合摄像头调整姿态,更无法控制光照环境。实际应用中,我们常…

2026/11/12 9:51:42

SSM电子书店源码全解析:结构、原理、部署与排错实战

/* MD / 富文本中的 .toc(含博客园搬家等嵌套结构);.toc-box 在侧栏,不受影响 */#content_views .toc,/* 编辑器常在目录前后插入空 p(:empty 仍占 20px),一并去掉避免顶空隙 */#content_views.markdown_views > p:empty:has(+ .toc),#content_views.markdown_views …

还想了解更多?直接咨询顾问

免费诊断 + 免费方案 + 透明报价。

全国咨询热线400-8866-253
免费获取方案
☎咨询二维码 ☎ ↑