Как загрузить предварительно обученную модель Iresnet100Python

Программы на Python
Anonymous
Как загрузить предварительно обученную модель Iresnet100

Сообщение Anonymous »

Нашел эти строки кода, но они, похоже, больше не работают. Кто-нибудь знает, каков обновленный синтаксис для загрузки модели iresnet100?
контекст: я редактирую существующий код Python, который использует предварительно обученная модель resnet50 model = models.resnet50(pretrained=True).cuda(). Мне нужно заменить его на модель Iresnet100

Код: Выделить всё

import insightface

model = insightface.iresnet100(pretrained=True)
model.eval()

Код: Выделить всё

AttributeError                            Traceback (most recent call last)
Cell In[28], line 2
1 import insightface
----> 2 embedder = insightface.iresnet100(pretrained=True)
3 embedder.eval()

AttributeError: module 'insightface' has no attribute 'iresnet100'

Код: Выделить всё

import torch
from insightface.model_zoo import get_model

embedder = get_model('buffalo_s', pretrained=True).cuda()  # Use the appropriate model name
embedder.eval()  # Set the model to evaluation mode
пробовал это, не помогло


Подробнее здесь: https://stackoverflow.com/questions/790 ... t100-model

Вернуться в «Python»