This repo is the official code for
python main.py
for training.Before you start your training, please make sure the dataset path is correct.
Check the dataset path in config.py
:
DIV_TRAIN_PATH = ''
DIV_VAL_PATH = ''
pub_TRAIN_PATH = ''
pub_VAL_PATH = ''
Here we provide a Demo.
You can hide a secret image into a host image with private key in our Demo by clicking your mouse.