Color-enhancement

Requirements

• CUDA 11.8
• CUDNN 8.9
• Python 3.9

Installation

• Make Conda Environment

conda create -n ColEn python=3.9 -y
conda activate ColEn

• Install Dependencies

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

conda install -c nvidia cudatoolkit=11.8 cudnn=8.9 -y

pip install -r requirements.txt

Download Pre-trained Models

Download pytorch models: [Pytorch model].

Download onnx models: [ONNX model].

Test

Run the following command for test:

• Pytorch model video test:

python pytorch_video_test.py --input<input video path> --output_dir<generated color enhance video path> --ckpt<pytorch color-enhancement model> --gpu_ids<ex)0,1,2,3>

• ONNX model video test:

python onnx_video_test.py --input<input video path> ---output_dir<generated color enhance video path> --onnx<onnx color-enhancement model> --gpu_ids<ex)0,1,2,3>