/SelfTMO

Primary LanguagePython

SelfTMO

This is the code of EG 22 paper: Learning a self-supervised tone mapping operator via feature contrast masking loss

Environment Configuration

conda env create -f environment.yaml

Online Training & Testing

Download the pre-trained VGG

  • download VGG weights VGG
python selftmo.py

You can change the 'data_dir' to your own testing images path

Offline Testing (Fast version)

python selftmo_offline.py

You can change the 'im_dir' to your wown testing images path