adaintest.py

def main():
    target = torch.randint(-20, 20, (8, 3, 4, 4)).float()
    source = torch.randint(-20, 20, (8, 3, 4, 4)).float()
 
    stylized_source = adain(source, target)
 
    target = target.view(8, 3, -1)
    stylized_source = stylized_source.view(8, 3, -1)
 
    # check variances the same
    target_variances = target.var(-1)
    stylized_variances = target.var(-1)
    print(torch.all(torch.abs(stylized_variances - target_variances) < 1e-6))
 
    # check means the same
    target_means = target.mean(-1)
    stylized_means = target.mean(-1)
    print(torch.all(torch.avs(stylized_means - target_means) < 1e-6))
 
    # check values are just rescaled / shifted
    print(torch.all(torch.abs(F.instance_norm(source) - F.instance_norm(stylized_source.view(8, 3, 4, 4))) < 1e-6))