Simple test time augmentation (TTA) for keras python library.
Switch branches/tags
Nothing to show
Clone or download
Latest commit d193ceb Aug 9, 2018
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
README.md Update README.md Aug 8, 2018
keras_tta.py Initial simple TTAModel_Wrapper commit Aug 8, 2018

README.md

keras_tta

Simple test time augmentation (TTA) for keras python library.

So far the wrapper flips the images horizontally and vertically and averages the predictions of all flipped images.

The intuition behind this is that even if the test image is not too easy to make a prediction, the transformations change it such that the model has higher chances of capturing the target shape and predicting accordingly.

Example:

tta_model = TTA_ModelWrapper(model)

predictions = tta_model.predict(X_test)