ImageClassification fails to load custom ResNet50 model with ".pt" file

[KaleBats123]

Environment Details;

• OS: Windows 11
• Python Version: 3.12
• ImageAI Version: 3.0.3
• PyTorch Version: 2.5.1+cu118

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;

1. Initialize ImageClassification
2. Set model path using setModelPath()
3. Set model type using setModelTypeAsResNet50()
4. 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?

[mario-s]

I get the same error when I use the example from the documentation

OS: macOS 15.4.1
Python: 3.13.0
ImageAI: 3.0.3
PyTorch: 2.7.0

Error:

RuntimeError: Cannot use ``weights_only=True`` with files saved in the legacy .tar format. In PyTorch 2.6, we changed the default value of the `weights_only` argument in `torch.load` from `False` to `True`. Re-running `torch.load` with `weights_only` set to `False` will likely succeed, but it can result in arbitrary code execution. Do it only if you got the file from a trusted source.
None
Weight loading failed.

predictions, probabilities = prediction.classifyImage(os.path.join(execution_path, "image1.jpg"), result_count=10)

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

[heyitsmass]

Same issue here, Was able to resolve by upgrading the version of torch to 2.5.1 in the requirements.txt file.
Environment Details;

OS: WSL 2 Ubuntu running on Windows 11
Python Version: 3.10
ImageAI Version: 3.0.3
PyTorch Version: 2.7.0

torch==2.5.1 --index-url //adjust for your use case
torchvision==0.20.1 --index-url //adjust for your use case

https://pytorch.org/get-started/previous-versions/