neural_spline_enhancement

neural_spline_enhancement copied to clipboard

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

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