The data generation folder contains all the scripts to generate the dataset and compute the mean and variance.
The Unet folder contains the codes for training and inferring the model.
The crop-warp folder contains the scripts for image warping.
The retrieval folder contains the code for image retrieval.