/neural_spline_enhancement

Personalization of the image enhancement style from few samples.

Primary LanguagePython

Implementation of the paper:

Personalized Image Enhancement using Neural Spline Color Transform
IEEE Transactions on Image Processing
S. Bianco, C. Cusano, F. Piccoli, R. Schettini

Pipeline

Inference

It is possible to replicate the results of the paper with the following command:

python regen.py \
--input_dir <DATASET_DIR>/raw \
--test_list <DATASET_DIR>/test-list.txt \
--out_dir ./regen \
--out_dir_splines ./regen_splines \
--model ./models/expC.pth \
--batchsize 10 \
--npoints 10 \
--nfilters 8 \
--nexperts 1