Skip to content

continue training on classification or transfer weight to expand dataset regulary #808

@dudutti

Description

@dudutti

Hi
to begin, great work, thanks.

I have a question to train model on classification, not detection.

If i expand my dataset when i add new photo but same number of class, i want to transfert learning like train_from_pretrained_model from custom Object detection but for Custom training prediction.
I've tried to pass in trainModel() :
model_trainer.trainModel(num_experiments=100, batch_size=32, train_from_pretrained_model=Myfilepath)

or i've tried .setTrainConfig() with the same result, in Custom training prediction these parameter doesn't exist

FIY : the documenation have mistake in https://imageai.readthedocs.io/en/latest/custom/index.html
the example of parameter in this code :
model_trainer.trainModel(num_objects=10, num_experiments=100, enhance_data=True, batch_size=32, show_network_summary=True)

trainModel() does not support anything than num_experiments and batch_size,
enhance_data=True and show_network_summary=True make an error
ClassificationModelTrainer.trainModel() got an unexpected keyword argument 'show_network_summary'

I use the lat version of ImageAI in repository by cloning with pip install imageai --upgrade v3.03

Many thanks, this question is just to retrained my model from the last instead of retrained all, it's a gain of 6 day of gpu computing with 1080TI

My project is to develop a program to mushroom classification, i have a dataset of 45 000 photo with 40 families and 100 species.
Everything work great but i need to expand my dataset regulary

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions