Ben Mildenhall | Jonathan T. Barron | Jiawen Chen | Dillon Sharlet | Ren Ng | Robert Carroll

This is not an official Google product. This repository contains code for training models from the paper Burst Denoising with Kernel Prediction Networks.

This code uses the following external packages:

• TensorFlow
• NumPy
• SciPy
• Matplotlib

Synthetic training data is generated using the OpenImages dataset, which can be manually downloaded following these instructions.

Run the following command to train the kernel prediction network (KPN) burst denoising model:

python kpn_train.py --dataset_dir $OPEN_IMAGES_DATASET_DIR --data_dir $REAL_BURST_DATA_DIR