This repo is the official code for
- PRIS: Practical robust invertible network for image steganography
- Python 3 (Recommend to use Anaconda).
- PyTorch = 1.11.0 .
- See requirements.txt for other dependencies.
- Run
python main.py
for training.
-
In this paper, we use the commonly used dataset DIV2K.
-
For train or test on your own dataset, change the code in
config.py
:line27: TRAIN_PATH = ''
line28: VAL_PATH = ''
-
Here we provide a Demo.
-
You can hide a secret image into a host image in our Demo by clicking your mouse.