Installation

Install via pip

pip install pytorchfwd

Usage

python -m pytorchfwd <path to dataset> <path to generated images>

Here are the other arguments and defaults used.

python -m pytorchfwd --help

usage: pytorchfwd.py [-h] [--batch-size BATCH_SIZE] [--num-processes NUM_PROCESSES] [--save-packets] [--wavelet WAVELET] [--max_level MAX_LEVEL] [--log_scale] path path

positional arguments:
  path                  Path to the generated images or path to .npz statistics file.

options:
  -h, --help            show this help message and exit
  --batch-size          Batch size for wavelet packet transform. (default: 128)
  --num-processes       Number of multiprocess. (default: None)
  --save-packets        Save the packets as npz file. (default: False)
  --wavelet             Choice of wavelet. (default: Haar)
  --max_level           wavelet decomposition level (default: 4)
  --log_scale           Use log scaling for wavelets. (default: False)
  --resize              Additional resizing. (deafult: None)

We conduct all the experiments with `Haar` wavelet with transformation/decomposition level of `4` for `256x256` image. The choice of max_level is dependent on the image resolution to maintain sufficient spial and frequency information. For 256 image-level 4, 128 image-level 3 and so on. In future, we plan to release the jax-version of this code.

Citation

If you use this work, please cite using following bibtex entry

@inproceedings{veeramacheneni25fwd,
author={Lokesh Veeramacheneni and Moritz Wolter and Hilde Kuehne and Juergen Gall},
title={Fréchet Wavelet Distance: A Domain-Agnostic Metric for Image Generation},
year={2025},
cdate={1735689600000},
url={https://openreview.net/forum?id=QinkNNKZ3b},
booktitle={ICLR},
crossref={conf/iclr/2025}}

Acknowledgments

The code is built with inspiration from Pytorch-FID. We use PyTorch Wavelet Toolbox for Wavelet Packet Transform implementation. We recommend to have a look at these repositories.

Testing

The tests folder contains tests to conduct independent verification of FWD. Github workflow executes all these tests. To run tests on your local system install nox, as well as this package via pip install ., and run

nox -s test

Funding

This research was supported by the Federal Ministry of Education and Research (BMBF) under grant no.01IS22094A WEST-AI and 6DHBK1022 BNTrAInee, the Deutsche Forschungsgemeinschaft (DFG, German Research Foundation) GA 1927/9-1 (KI-FOR 5351) and the ERC Consolidator Grant FORHUE (101044724). Prof. Kuehne is supported by BMBF project STCL - 01IS22067. The authors gratefully acknowledge the Gauss Centre for Supercomputing e.V.(www.gauss-centre.eu) for funding this project by providing computing time through the John von Neumann Institute for Computing (NIC) on the GCS Supercomputer JUWELS at Jülich Supercomputing Centre (JSC). The authors heartfully thank all the volunteers who participated in the user study. The sole responsibility for the content of this publication lies with the authors.