I want to create a custom model that improves the color and tone of the image

Image-Adaptive-3DLUT is a model that can learn 3D Look Up Table to enhance the image.

The pre-trained model is trained on a retouched dataset called FiveK, but it can also be trained on custom data.

Training procedure

Data preparation

Prepare the data in the following format.

Resized to 480px square.

set up