End to End Image Classifier

Params file

Edit for training and model parameters like epoch, batch size, split ratio e.t.c

Training

python train.py --model_dir path_to_param_file

path to param file argument may be left blank if fiile in the same directory as train.py

Inference

python predict.py --img_dir path_to_images_folder