Как загрузить предварительно обученную модель 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
Реклама
Ответить Пред. темаСлед. тема

Быстрый ответ

Изменение регистра текста: 
Смайлики
:) :( :oops: :roll: :wink: :muza: :clever: :sorry: :angel: :read: *x)
Ещё смайлики…
   
К этому ответу прикреплено по крайней мере одно вложение.

Если вы не хотите добавлять вложения, оставьте поля пустыми.

Максимально разрешённый размер вложения: 15 МБ.

  • Похожие темы
    Ответы
    Просмотры
    Последнее сообщение

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