KO | EN

GitLite

Minimalist Open Source Repository Search Engine

gitlite — search

#dsh #dsh-plugin #ai-agents #developer-tools #typescript #codex #cli #claude-code #mcp #rust #deepseek-harness #ai-agent

ICELUT

★ 51 Open GitHub ↗

[ECCV 2024] Taming Lookup Tables for Efficient Image Retouching

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

#  [ECCV2024] Taming Lookup Tables for Efficient Image Retouching

[Paper](https://arxiv.org/pdf/2403.19238.pdf)

:star: If ICELUT is helpful to your images or projects, please help star this repo. Thanks! :hugs:

## TODO
- [x] Add training code and config files
- [x] Add LUT and inference code
## Requirements and Dependencies
+ python==3.7
+ cuda==11.1
+ numpy==1.21.5
+ torch==1.8.0
+ torchvision==0.9.0
+ opencv-python==4.5.5.62
+ matplotlib==3.5.1
+ scipy==1.7.3
+ GPU: NVIDIA GeForce RTX 3090
+ CPU: Intel(R) Xeon(R) Platinum

## How to run the codes

### 1. Install the tilinear interpolation package

#### 1.1 For GPU User
+ Check your cuda version
```bash
    $ls /usr/local/
```
+ Change the cuda path in your setting
```bash
    $cd trilinear_cpp
    $vim setup.sh
```
```bash
    export CUDA_HOME=/usr/local/your_cuda_version && python setup.py install
```
+ Install the package
```python
    $sh setup.sh
```
#### 1.2 For CPU User

+ Install the package
```bash
    $cd trilinear_cpp
    $python setup.py
```
**ATTENTION:** If you follow the CPU install instruction with the GPU in your device, the default programme will still install the GPU version. If you want to only install the CPU version, please follow this step:
+ Check the setup.py codes
```bash
    $vim setup.py
```

+ Substitute the codes
```python
    # line 5
    if torch.cuda.is_available():

# substitute line 5 with :
    # if False
```

### 2. Inference the demo images
```bash
    $python inference_demo.py
```

### 3. Training 
#### Prepare the datasets
Please refer to [3D LUT](https://github.com/HuiZeng/Image-Adaptive-3DLUT) to prepare the FiveK datasets.
#### Training model
```bash
    $python train_ICELUT.py
```

### 4. Transfer to LUTs
```bash
    $python transfer2LUT.py
```

### 5. Optional: Check the retouched results
The retouched images generated by running inference_demo.py will be saved in the default dir: ./test_image_output

### 6. Optional: Check the total LUT size
```bash
cd ./ICELUT
du -sh *.npy
```

```bash
% output
52K     Basis_lut.npy
204K    classifier_int8.npy
164K    Model_lsb_fp32.npy
164K    Model_msb_fp32.npy
```
Note that we use 10 Basis in codes rather than 20 in the paper. So the storage could be smaller (588KB v.s. 780KB).

## Acknowledgement
This project is based on [CLUT](https://github.com/Xian-Bei/CLUT). Thanks for these awesome codes!

ICELUT

Explore Similar Repositories

Was this content helpful?

Recent Feedback

Download README