如何加载预训练的iresnet100模型

Question

找到这些代码行，但它们似乎不再起作用，有人知道加载 iresnet100 模型的更新语法是什么？上下文是我正在编辑一个使用 resnet50 预训练模型“model = models.resnet50(pretrained=True).cuda()”的现有 python 代码。我需要用 iresnet100 型号替换它

`导入insightface

模型= Insightface.iresnet100（预训练= True） 模型.eval()

`

AttributeError Traceback（最近一次调用最后一次） 单元格 In[28]，第 2 行 1导入insightface ----> 2 嵌入器 = Insightface.iresnet100(pretrained=True) 3 嵌入器.eval()

属性错误：模块“insightface”没有属性“iresnet100”

`进口火炬 从 Insightface.model_zoo 导入 get_model

embedder = get_model('buffalo_s', pretrained=True).cuda() # 使用适当的模型名称 embedder.eval() # 将模型设置为评估模式 `

尝试过这些，但没有成功

Answer 1

该问题可能是由于

insightface

库 API 的更改造成的。要使用更新的语法加载

iresnet100

模型，您可以尝试：

from insightface.model_zoo import get_model

model = get_model('arcface_r100_v1', download=True)
model.eval()

这里，

'arcface_r100_v1'

相当于

iresnet100

模型。如果您使用 CUDA，您可以将模型移动到 GPU：

model.cuda()

另外，准备模型：

model.prepare(ctx_id=0)