Metadata-Version: 2.4
Name: synthpix
Version: 0.1.2
Summary: A lightspeed PIV images generator using JAX
Author-email: Antonio Terpin <aterpin@ethz.ch>, Alan Bonomi <abonomi@ethz.ch>, Francesco Banelli <fbanelli@ethz.ch>
License: MIT
Project-URL: Homepage, https://github.com/antonioterpin/synthpix
Project-URL: Repository, https://github.com/antonioterpin/synthpix
Project-URL: Documentation, https://github.com/antonioterpin/synthpix#readme
Project-URL: Issues, https://github.com/antonioterpin/synthpix/issues
Keywords: PIV,synthetic data,jax
Classifier: Development Status :: 4 - Beta
Classifier: Intended Audience :: Science/Research
Classifier: Intended Audience :: Developers
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Topic :: System :: Logging
Requires-Python: >=3.10
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: jax>=0.6.0
Requires-Dist: tqdm>=4.67.1
Requires-Dist: h5py>=3.13.0
Requires-Dist: ruamel.yaml>=0.18.10
Requires-Dist: Pillow>=10.0.0
Requires-Dist: robo-goggles>=0.1.7
Requires-Dist: numpy>=2.2.6
Requires-Dist: gdown>=5.2.0
Requires-Dist: grain>=0.2.13
Requires-Dist: orbax-checkpoint>=0.11.31
Provides-Extra: dev
Requires-Dist: pytest>=7.0.0; extra == "dev"
Requires-Dist: pytest-cov>=4.0.0; extra == "dev"
Requires-Dist: hypothesis>=6.0.0; extra == "dev"
Requires-Dist: snowballstemmer==2.2.0; extra == "dev"
Requires-Dist: pre_commit==4.0.1; extra == "dev"
Requires-Dist: tomli; extra == "dev"
Requires-Dist: pyupgrade>=3.19.0; extra == "dev"
Requires-Dist: matplotlib>=3.10.1; extra == "dev"
Requires-Dist: ruff>=0.9.0; extra == "dev"
Requires-Dist: mypy>=1.14.0; extra == "dev"
Provides-Extra: cuda12
Requires-Dist: jax[cuda12]>=0.6.0; extra == "cuda12"
Provides-Extra: download
Requires-Dist: gdown>=5.1.0; extra == "download"
Requires-Dist: scipy>=1.10.0; extra == "download"
Requires-Dist: tensorflow-cpu>=2.15.0; extra == "download"
Dynamic: license-file

# SynthPix: A lightspeed PIV images generator 🌊

[![GitHub stars](https://img.shields.io/github/stars/antonioterpin/synthpix?style=social)](https://github.com/antonioterpin/synthpix/stargazers)
[![License: MIT](https://img.shields.io/badge/License-MIT-green.svg)](https://creativecommons.org/licenses/by-sa/4.0/)
[![codecov](https://codecov.io/gh/antonioterpin/synthpix/graph/badge.svg?token=UQ48NNZSI4)](https://codecov.io/gh/antonioterpin/synthpix)
[![Tests](https://github.com/antonioterpin/synthpix/actions/workflows/test.yaml/badge.svg)](https://github.com/antonioterpin/synthpix/actions/workflows/test.yaml)
[![PyPI version](https://img.shields.io/pypi/v/synthpix.svg)](https://pypi.org/project/synthpix)

`SynthPix` is a synthetic image generator for Particle Image Velocimetry (PIV) with a focus on performance and parallelism on accelerators, implemented in [JAX](https://docs.jax.dev/en/latest/quickstart.html). `SynthPix` supports the same configuration parameters as existing tools but achieves a throughput several orders of magnitude higher in image-pair generation per second, enabling comprehensive validation and comparison of PIV algorithms, rapid experimental design iterations, and the development of data-hungry methods.
![The SynthPix image generation pipeline](https://raw.githubusercontent.com/antonioterpin/synthpix/main/docs/synthpix.jpg)

In a nutshell, if you need many synthetic PIV images and you do not want to wait ages, you are better off with `SynthPix` 😄. Below are the performances (image pairs per second) with and without GPU for different batch sizes B.
![Performances](https://raw.githubusercontent.com/antonioterpin/synthpix/main/docs/times.jpg)

`SynthPix` is also fairly easy to use:
```python
import synthpix

sampler = synthpix.make(config_path)
for i, batch in enumerate(sampler):
   """batch contains images1, images2, flow_fields, params"""
sampler.shutdown()
```
See `src/main.py` for a fully working example. For long-running training, check out our [Checkpointing Guide](docs/checkpointing.md) to ensure bit-perfect reproducibility 💾.

## Getting started 🚀
Alright, now that hopefully we convinced you to try SynthPix, let's get to it. Don't worry, installing it is even easier than using it:
```bash
uv add synthpix
```
If you have CUDA GPUs,
```bash
uv add "synthpix[cuda12]"
```
If you have issues with CUDA drivers, please follow the official instructions for [cuda12](https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=22.04&target_type=deb_local) and [cudnn](https://developer.nvidia.com/cudnn-downloads?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=22.04&target_type=deb_local)
(*Note: wheels only available on linux*). Make sure that `LD_LIBRARY_PATH` is not set; see the [JAX docs](https://docs.jax.dev/en/latest/installation.html):
```bash
unset LD_LIBRARY_PATH
```

Check out our [instructions](./CONTRIBUTING.md) for installing `SynthPix` from source. You can of course also use pip.

To generate the images, you need flow data. We provide two scripts to download the commonly used PIV datasets:
```bash
sh scripts/download_piv_1.sh <output_folder> <configs_directory>
```
```sh
sh scripts/download_piv_2.sh <output_folder> <configs_directory>
```
These scripts will automatically save also one configuration file each in `<configs_directory>`. You can use these paths as in the example above.

For more examples and tutorials to use custom flow data or real-world data, check out our [tutorials page](docs/tutorials.md).

## Configuring the synthetic images ⚙️

Configuration is handled through a YAML file, organized into four main groups. Here’s a quick guide to what each set of parameters does:

### 1. Dataset Parameters
These parameters allow you to customize how flow field extraction works and the shape of the output batch.

| **Parameter**            | **Description**                         |
| ------------------------ | --------------------------------------- |
| `seed`                   | Random seed for reproducibility         |
| `batch_size`             | Number of image pairs generated at once |
| `flow_fields_per_batch`  | Number of unique flow fields per batch  |
| `batches_per_flow_batch` | Number of batches generated before updating to a new set of flow fields |

### 2. Image Generation Parameters
Define the look and realism of your synthetic PIV images.

| **Parameter**                 | **Description**                                             |
| ----------------------------- | ----------------------------------------------------------- |
| `image_shape`                 | Output image size `[height, width]` (pixels)                |
| `dt`                          | Time between frames (seconds)                               |
| `seeding_density_range`       | Range of particle densities (particles per pixel)           |
| `diameter_ranges`             | List of possible ranges for particle sizes (in pixels). (see below)     |
| `diameter_var`                | Variability in particle size                                |
| `intensity_ranges`            | List of possible intensity (brightness) ranges. (see below)             |
| `intensity_var`               | Variability in intensity                                    |
| `p_hide_img1` / `p_hide_img2` | Per-particle probability of being hidden in image 1 (or 2). |
| `rho_ranges`                  | List of possible correlation coefficients ranges. (see below)            |
| `rho_var`                     | Variability in correlation                                  |
| `noise_uniform`                 | Maximum amplitude of uniform noise added to the image.|
| `noise_gaussian_mean`                 | Mean of Gaussian noise added to the image.             |
| `noise_gaussian_std`                 | Standard deviation of Gaussian noise added to the image.           |
| `mask` (optional)                |  Path to a `.npy` file containing a binary mask (0 and 1) with shape `image_shape`.|
| `histogram` (optional)                |  Path to a `.npy` file with a 1D array of shape `(256,)`, summing to the number of image pixels. Used to remap output intensities.           |

For `diameter_ranges`, `intensity_ranges`, and `rho_ranges`, each parameter is a list of possible ranges. For every image pair, one range is randomly selected (uniformly) from each list, and then the particle parameters are sampled from these selected ranges.

Each *_var parameter controls how much the property of each particle can change between the first and second image of a pair. The change is applied as Gaussian noise with zero mean and variance given by the corresponding *_var value.

Here’s what each *_var parameter controls:

**diameter_var:**
Simulates changes in particle size between frames, making the effect of focus, depth movement, or slight deformations more realistic.

Effect of `diameter_var` on particle size
<p align="center" style="margin-bottom:0;">
  <span style="display:inline-block; width: 90px; text-align:center;"><b>Low</b></span>
  <span style="display:inline-block; width: 90px; text-align:center;"><b>Medium</b></span>
  <span style="display:inline-block; width: 90px; text-align:center;"><b>High</b></span>
</p>
<p align="center" style="margin-top:0;">
  <img src="https://raw.githubusercontent.com/antonioterpin/synthpix/main/docs/images/diam2.png" width="90" style="image-rendering: pixelated; margin: 0 10px; vertical-align: middle;"/>
  <img src="https://raw.githubusercontent.com/antonioterpin/synthpix/main/docs/images/base.png" width="90" style="image-rendering: pixelated; margin: 0 10px; vertical-align: middle;"/>
  <img src="https://raw.githubusercontent.com/antonioterpin/synthpix/main/docs/images/diam3.png" width="90" style="image-rendering: pixelated; margin: 0 10px; vertical-align: middle;"/>
</p>
<p align="center" style="margin-top:0;">
  <span style="display:inline-block; width: 90px; text-align:center;">diameter = 2.0</span>
  <span style="display:inline-block; width: 90px; text-align:center;">diameter = 2.5</span>
  <span style="display:inline-block; width: 90px; text-align:center;">diameter = 3.0</span>
</p>
<p align="center">
  <sub>
    Increasing <code>diameter_var</code> makes particles appear larger and more diffuse.
  </sub>
</p>

**intensity_var**:
Models natural brightness changes due to variations in illumination, camera response, or particles moving in and out of the light sheet mimicking out-of-plane motion.

Effect of `intensity_var` on particle brightness
<p align="center" style="margin-bottom:0;">
  <span style="display:inline-block; width: 90px; text-align:center;"><b>Low</b></span>
  <span style="display:inline-block; width: 90px; text-align:center;"><b>Medium</b></span>
  <span style="display:inline-block; width: 90px; text-align:center;"><b>High</b></span>
</p>
<p align="center" style="margin-top:0;">
  <img src="https://raw.githubusercontent.com/antonioterpin/synthpix/main/docs/images/50_Int.png" width="90" style="image-rendering: pixelated; margin: 0 10px; vertical-align: middle;"/>
  <img src="https://raw.githubusercontent.com/antonioterpin/synthpix/main/docs/images/150_Int.png" width="90" style="image-rendering: pixelated; margin: 0 10px; vertical-align: middle;"/>
  <img src="https://raw.githubusercontent.com/antonioterpin/synthpix/main/docs/images/base.png" width="90" style="image-rendering: pixelated; margin: 0 10px; vertical-align: middle;"/>
</p>
<p align="center" style="margin-top:0;">
  <span style="display:inline-block; width: 90px; text-align:center;">intensity = 50</span>
  <span style="display:inline-block; width: 90px; text-align:center;">intensity = 150</span>
  <span style="display:inline-block; width: 90px; text-align:center;">intensity = 255</span>
</p>
<p align="center">
  <sub>
    Raising the <code>intensity_var</code> parameter produces brighter particles (higher signal-to-noise).
  </sub>
</p>


**rho_var**:
Adds variability to the shape and orientation (elongation/rotation) of particles between frames.

Effect of `rho_var` on particle shape:
<p align="center" style="margin-bottom:0;">
  <span style="display:inline-block; width: 90px; text-align:center;"><b>Too Low</b></span>
  <span style="display:inline-block; width: 90px; text-align:center;"><b>Negative</b></span>
  <span style="display:inline-block; width: 90px; text-align:center;"><b>Zero</b></span>
  <span style="display:inline-block; width: 90px; text-align:center;"><b>Positive</b></span>
  <span style="display:inline-block; width: 90px; text-align:center;"><b>Too High</b></span>
</p>
<p align="center" style="margin-top:0;">
  <img src="https://raw.githubusercontent.com/antonioterpin/synthpix/main/docs/images/rho_low.png" width="90" style="image-rendering: pixelated; margin: 0 10px; vertical-align: middle;"/>
  <img src="https://raw.githubusercontent.com/antonioterpin/synthpix/main/docs/images/base.png" width="90" style="image-rendering: pixelated; margin: 0 10px; vertical-align: middle;"/>
  <img src="https://raw.githubusercontent.com/antonioterpin/synthpix/main/docs/images/no_rho.png" width="90" style="image-rendering: pixelated; margin: 0 10px; vertical-align: middle;"/>
  <img src="https://raw.githubusercontent.com/antonioterpin/synthpix/main/docs/images/rho_positive.png" width="90" style="image-rendering: pixelated; margin: 0 10px; vertical-align: middle;"/>
  <img src="https://raw.githubusercontent.com/antonioterpin/synthpix/main/docs/images/rho_high.png" width="90" style="image-rendering: pixelated; margin: 0 10px; vertical-align: middle;"/>
</p>
<p align="center" style="margin-top:0;">
  <span style="display:inline-block; width: 90px; text-align:center;">ρ = -0.9</span>
  <span style="display:inline-block; width: 90px; text-align:center;">ρ = -0.5</span>
  <span style="display:inline-block; width: 90px; text-align:center;">ρ = 0.0</span>
  <span style="display:inline-block; width: 90px; text-align:center;">ρ = 0.5</span>
  <span style="display:inline-block; width: 90px; text-align:center;">ρ = 0.9</span>
</p>
<p align="center">
  <sub>
    Increasing the absolute value of <code>rho</code> (correlation coefficient) makes particle spots more elliptical and/or rotated.<br>
    Very high values (|ρ| &gt; 0.5, leftmost and rightmost) lead to elongated, line-like particles that are not realistic for PIV images.<br>
    For best results, keep <code>rho</code> in the range <b>-0.5 ≤ ρ ≤ 0.5</b>.
  </sub>
</p>


  Note: The example above uses particles with diameter 2.5. The visual effect of ρ also depends on particle size, larger diameters make elongation more noticeable.
  As a result, the recommended "safe range" for realistic particle shapes may vary depending on your chosen `diameter_ranges`.

### 3. Flow generation parameters
| **Parameter**          | **Description**                          |
| ---------------------- | ---------------------------------------- |
| `velocities_per_pixel` | Spatial resolution of the velocity field |
| `resolution`           | Pixels per unit of physical length       |

#### Examples:

| **Use Case**                | `velocities_per_pixel` | `resolution` | Explanation                                                                                                       |
| --------------------------- | ---------------------- | ------------ | ----------------------------------------------------------------------------------------------------------------- |
| **Normalized flow field**   | 1.0                    | 100          | Flow field has 1 velocity vector per pixel; 100 px = 1 unit of length and flow velocity vectors are in units/s.                          |
| **Real-world flow in m/s**  | 10.0                   | 50           | Flow field has 10 velocity vectors per image pixel (very fine spatial resolution); 50 px = 1 meter and flow in m/s. |
| **Pixel displacement flow** | 1.0                    | 1.0          | Flow field has 1 velocity per pixel; 1 px = 1 unit → flow directly describes pixels/second displacements. (use `dt` = 1 to convert it in pixels/frame)           |




### 4. Flow field parameters
Describe which region of your flow field is captured, and its velocity bounds.

| **Parameter**                    | **Description**                                         |
| -------------------------------- | ------------------------------------------------------- |
| `flow_field_size`                | Physical size of the entire flow field (e.g., in mm × mm). Units must match those in your flow files. |
| `img_offset` | 2D offset (in physical units, matching `flow_field_size`) specifying the top-left corner of the region of interest to extract from the flow field. |
| `min_speed_x/y`, `max_speed_x/y` | Range of allowed velocities in each direction,           |
| `output_units`                   | Units used in the returned flow field: `"pixels"` (converts physical velocities to displacements in pixels using `dt` and `resolution`), or `"measure units per second"` (same units as the input flow field, e.g., mm/s). |
| `file_list`                | List of ground-truth flow field files (e.g. `.mat`)      |
| `scheduler_class`                | Loader class for your flow files (usually by extension). See [tutorials](docs/tutorials.md) for detailed explanations. |

Note: `min_speed_x/y` and `max_speed_x/y` are not absolute cutoffs, but define the full expected velocity range (positive and negative) along each axis.

These bounds are used to estimate the maximum particle displacement over time (`dt`). A larger intermediate image is generated accordingly, and the final image pair is cropped from this larger region. This ensures that particles entering the visible frame (image 2) have realistic origins — possibly outside the region of interest in image 1 — making particle "appearance" at the boundaries realistic.

## Contributing 🤗
Contributions are more than welcome! 🙏 Please check out our [how to contribute page](./CONTRIBUTING.md), and feel free to open an issue for problems and feature requests⚠️.

## Citation 📈
If you use this code in your research, please cite our paper:
```bash
   @misc{terpin2025synthpixlightspeedpivimages,
        title={SynthPix: A lightspeed PIV images generator},
        author={Antonio Terpin and Alan Bonomi and Francesco Banelli and Raffaello D'Andrea},
        year={2025},
        eprint={2512.09664},
        archivePrefix={arXiv},
        primaryClass={cs.DC},
        url={https://arxiv.org/abs/2512.09664},
    }
```
