Skip to content

Latest commit

 

History

History
24 lines (19 loc) · 945 Bytes

readme.md

File metadata and controls

24 lines (19 loc) · 945 Bytes

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