generic_image_classifier

An easy and clean repository to perform image classification tasks.

Models Covered

So far four models are available to train namely:

Place the dataset in the folder and set the path_dataset variable to that folder. Do the same with path_validation.
Set validation steps to according to your dataset size.
Set the learning rate.
Set the batch size according to you GPU memory.
Select the model you want to train via model_num.
You can also turn on the augmentor in the datagen.py file.
run train.py
After the model is trained run inference.py file to get all the plots and results.