# Model zoo¶

In our experiments, we re-trained a set of models with the harmonization loss proposed in the paper. You can easily download the weights of each models here:

In order to load them easily, we have set up utilities in the github repository. For example, to load the model lives harmonized:

from harmonization.models import (load_ViT_B16, load_ResNet50,
preprocess_input)

# load images (in [0, 255])
# ...

images = preprocess_input(images)
predictions = vit_harmonized(images)