You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Problem Description;
When attempting to load a custom-trained ResNet50 model using ImageAI's ImageClassification, the model fails to load with the error:
"Classification failed: Model not yet loaded. You need to call '.loadModel()' before performing image classification"
Reproduction Steps;
Initialize ImageClassification
Set model path using setModelPath()
Set model type using setModelTypeAsResNet50()
Call loadModel()
Code Snippet;
from imageai.Classification import ImageClassification
classifier = ImageClassification()
classifier.setModelPath("path/to/model.pt")
classifier.setModelTypeAsResNet50()
classifier.loadModel() # Fails here
Additional Context;
Model trained using custom dataset
Model saved as '.pt' file
Attempting to use for image classification
Questions;
Does ImageAI support loading custom-trained models?
Are there specific model export requirements?
The text was updated successfully, but these errors were encountered:
Environment Details;
Problem Description;
When attempting to load a custom-trained ResNet50 model using ImageAI's
ImageClassification
, the model fails to load with the error:"Classification failed: Model not yet loaded. You need to call '.loadModel()' before performing image
classification"
Reproduction Steps;
setModelPath()
setModelTypeAsResNet50()
loadModel()
Code Snippet;
Additional Context;
Model trained using custom dataset
Model saved as '.pt' file
Attempting to use for image classification
Questions;
Does ImageAI support loading custom-trained models?
Are there specific model export requirements?
The text was updated successfully, but these errors were encountered: