Hans1984 / SelfTMO

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 own testing images path