python调用pytorch的clip模型时报错-Toy模板网

这篇具有很好参考价值的文章主要介绍了python调用pytorch的clip模型时报错。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

使用python调用pytorch中的clip模型时报错：AttributeError: partially initialized module ‘clip’ has no attribute ‘load’ (most likely due to a circular import)

现象

clip源码及官方文档

在使用pyton调用pytorch中的clip模型获取图片的关键词时，根据官方文档的代码描述，代码内容如下

import torch
import clip
from PIL import Image

device = "cuda" if torch.cuda.is_available() else "cpu"
model, preprocess = clip.load("ViT-B/32", device=device)

image = preprocess(Image.open("C:/Users/HLY/Desktop/3.jpeg")).unsqueeze(0).to(device)
text = clip.tokenize(["a diagram", "a dog", "a cat"]).to(device)

with torch.no_grad():
    image_features = model.encode_image(image)
    text_features = model.encode_text(text)

    logits_per_image, logits_per_text = model(image, text)
    probs = logits_per_image.softmax(dim=-1).cpu().numpy()

print("Label probs:", probs)  # prints: [[0.9927937  0.00421068 0.00299572]]

根据官方文档安装pytorch，clip等版本后，报错，报错信息如下

Traceback (most recent call last):
  File "D:\WorkTest\pythonProject\resnetTest\clip.py", line 2, in <module>
    import clip
  File "D:\WorkTest\pythonProject\resnetTest\clip.py", line 6, in <module>
    model, preprocess = clip.load("ViT-B/32", device=device)
                        ^^^^^^^^^
AttributeError: partially initialized module 'clip' has no attribute 'load' (most likely due to a circular import)