Skip to content

The goal of this repo is to show how to perform TTA, a simple technique to improve the predictions of your model, with Keras

Notifications You must be signed in to change notification settings

nathanhubens/TTA-Keras

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 

Repository files navigation

TTA-Keras

Test-Time Augmentation is a very efficient way to improve the results of your model at testing time.

Here is how it works for a single image

After having trained our network, we can show it a test image and see it prediction.

Here is the query image :

Optional Text

Its prediction is :

Optional Text

While the true label is :

Optional Text

So the test image is misclassified.

But what happens if we apply TTA to that image ?

Meaning that we will show the model several slightly modified versions of the image and average its predictions.

The 5 modified images are :

Optional Text

And the corresponding predictions :

  • Prediction 1: Optional Text

  • Prediction 2: Optional Text

  • Prediction 3: Optional Text

  • Prediction 4: Optional Text

  • Prediction 5: Optional Text

By now taking the average we have: Optional Text

And the image is correctly classified !

This notebook shows how to use it with Keras

Further explanation can be found here

About

The goal of this repo is to show how to perform TTA, a simple technique to improve the predictions of your model, with Keras

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published