image-classifier

Usage

To try the default classifier model on unsplit directory of subdir-labelled images:

python main.py --image_dir "<sub-dir-labeled-image-dir>" \
--logdir "<output_dir>"

To try the default classifier model on directory of subdir-labelled images already split into train/test or train/test/val directories:

python main.py --split_image_dir "<sub-dir-labeled-split-image-dir>" \
--logdir "<output_dir>"

To try many models

formin$(python available_tf_hub_models.py);do python main.py --image_dir "<sub-dir-labeled-image-dir>" \
--logdir "<output_dir>" \
--run_name "$m" \
--model "$m"done