search

OlafenwaMoses / ImageAI

A python library built to empower developers to build applications and systems with self-contained Computer Vision capabilities
https://www.genxr.co/#products
MIT License
8.65k stars 2.19k forks source link

TypeError #829

Open CodeWithChai opened 1 year ago

CodeWithChai commented 1 year ago

from imageai.Classification.Custom import ClassificationModelTrainer model_trainer = ClassificationModelTrainer() model_trainer.setModelTypeAsResNet50() model_trainer.setDataDirectory(r"/content/drive/MyDrive/pets") model_trainer.trainModel(num_objects=10, num_experiments=100, enhance_data=True,batch_size=32, show_network_summary=True)

-- I keep on getting this error despite trying everything, any solutions ? TypeError: ClassificationModelTrainer.trainModel() got an unexpected keyword argument 'num_objects'