This repo is the official code for
- Python 3 (Recommend to use Anaconda).
- PyTorch = 1.11.0 .
- See requirements.txt for other dependencies.
- Run
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.