Color Style Transfer with Modulated Flows

This is the source code of the paper "Color Style Transfer with Modulated Flows" for Workshop SPIGM @ ICML 2024.

Please refer to the

• src directory for models definitions
• generate_flows_v2 script for training the dataset of rectified flows
• train_encoder_v2 script for training the encoder

Call python3 run_inference.py --help to see a full list of arguments for inference. Ctrl+C cancels the execution.