Как загрузить специально обученную модель ResNet50 с помощью ImageAI в Python? ⇐ Python

[Anonymous]

Я работаю над проектом классификации изображений, используя специально обученную модель ResNet50 с ImageAI, но сталкиваюсь с постоянной ошибкой загрузки, которая не позволяет мне использовать обученную модель для прогнозов. Эта ошибка предполагает, что существует проблема совместимости или какой-то конкретный метод, который мне не хватает при попытке загрузить специально обученную модель PyTorch в среду ImageAI.
Я попытался загрузить специально обученную ResNet50. модель с использованием стандартной процедуры загрузки модели ImageAI:

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

from imageai.Classification import ImageClassification

classifier = ImageClassification()
classifier.setModelPath("path/to/custom/model.pt")
classifier.setModelTypeAsResNet50()
classifier.loadModel()  # Expected to successfully load the model

Я ожидал, что процесс загрузки модели будет простым, похожим на использование предварительно обученных моделей. Вместо этого я получил досадную ошибку:

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

Classification failed: Model not yet loaded. You need to call '.loadModel()'
before performing image classification

Сведения о моей среде:

• Python 3.12
• ImageAI 3.0.3
• PyTorch 2.5.1+cu118
• Windows 11

Я подтвердил:

• Путь к модели указан правильно.
• Модель сохранена как действительный файл .pt.
• модель была обучена на специальном наборе данных с использованием архитектуры ResNet50.

Конкретные вопросы:

• Существует ли специальный метод загрузки специально обученных моделей в ImageAI?
• Существуют ли какие-либо особые требования к экспорту или преобразованию для пользовательских моделей?
• Нужно ли мне изменить словарь состояний моей модели или использовать определенный формат экспорта?

Подробнее здесь: https://stackoverflow.com/questions/792 ... -in-python