使用huggingface模型时报错LocalEntryNotFoundError

在使用huggingface上的模型facebook/detr-resnet-50时,报错:
在这里插入图片描述
但是我的代码是这么写的:

from transformers import DetrImageProcessor, DetrForObjectDetection
import torch
from PIL import Image

image = Image.open("image/000000039769.jpg")

processor = DetrImageProcessor.from_pretrained("model_zoo/detr-resnet-50", local_files_only=True)
model = DetrForObjectDetection.from_pretrained("model_zoo/detr-resnet-50", local_files_only=True)

inputs = processor(images=image, return_tensors="pt")
outputs = model(**inputs)

target_sizes = torch.tensor([image.size[::-1]])
results = processor.post_process_object_detection(outputs, target_sizes=target_sizes, threshold=0.9)[0]

for score, label, box in zip(results["scores"], results["labels"], results["boxes"]):
    box = [round(i, 2) for i in box.tolist()]
    print(
            f"Detected {model.config.id2label[label.item()]} with confidence "
            f"{round(score.item(), 3)} at location {box}"
    )

已经将模型下载到本地了,感觉没有用到什么网络,不应该会报这里的错误,仔细检查后,发现是这里:
在这里插入图片描述
也就是DETR会使用一个resnet50,在初始化resnet50的时候出现了问题,于是先到官网将对应的参数下载到本地。
在这里插入图片描述
然后调整这里的函数,
在这里插入图片描述
在其中额外添加参数pretrained_cfg_overlay即可。

backbone = create_model(
	config.backbone,
	pretrained=config.use_pretrained_backbone,
	features_only=True,
	out_indices=(1, 2, 3, 4),
	in_chans=config.num_channels,
    pretrained_cfg_overlay=dict(file='your_path/pytorch_model.bin'),
	**kwargs
)

finally,成功!
在这里插入图片描述

使用 Hugging Face 模型,你可以按照以下步骤进行: 1. 在 Hugging Face 模型库(https://huggingface.co/models)中选择你需要的预训练模型,并下载模型文件。你可以从模型文件中获取所需的框架(如 TensorFlow、PyTorch)模型文件和配置文件等。 2. 了解模型的功能和性能。你可以查看模型介绍(Model Card)文档,该文档提供了模型的详细信息,包括模型的功能和性能。 3. 使用提供的代码样例(Use in Transformers)来了解如何使用模型。你可以直接拷贝代码到你的项目中,并按照示例进行修改和使用。 4. 如果需要测试模型,你可以使用模型的测试模块(Hosted inference API)。这使你能够直接在该模块上测试你自己的数据,同 Hugging Face 还提供了 Http API 可以调用,这样就不需要本地部署了。 5. 如果 Hugging Face 提供的模型无法满足你的需求,你可以进行自己的模型训练。在这种情况下,你可以使用 Hugging Face 提供的预训练模型进行迁移学习。详情请参考 Hugging Face 提供的迁移学习文档,了解如何使用预训练模型进行迁移学习。 总结来说,你可以通过在 Hugging Face 模型库中选择和下载预训练模型,查看模型介绍和代码样例,使用模型的测试模块进行测试,并根据需要进行自己的模型训练和迁移学习。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* [hugging face 模型库的使用及加载 Bert 预训练模型](https://blog.csdn.net/IT__learning/article/details/120741368)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"] - *2* *3* [Hugging Face快速入门(重点讲解模型(Transformers)和数据集部分(Datasets))](https://blog.csdn.net/zhaohongfei_358/article/details/126224199)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值