Data preparation:

-gen_trainingpaches.py: script to automatically tile big images into small patches for training.

usage:

python gen_trainingpatches.py --root DIR_TO_ORIGINAL_IMAGES --step STEP_SIZE --size SIZE_OF_PATCHES --output OUTPUT_DIR

The folder “A” that contains the training dataset should be organized as follow:

A\train\whales contains all training images for whale

A\train\water contains all training images for water

Training:

python transfer_learning.py --name MODEL_NAME --data_dir TRAINING_DATA_DIR