контекст: я редактирую существующий код 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