DKiS: Decay weight invertible image steganography with private key

This repo is the official code for

• DKiS: Decay weight invertible image steganography with private key

Dependencies and Installation

• Python 3 (Recommend to use Anaconda).
• PyTorch = 1.11.0 .
• See requirements.txt for other dependencies.

Get Started

• Run python main.py for training.

Dataset

• 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 = ''

Demo

• 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.

Pre-trained model state

• "model_state.pt" in "final_state" folder is a pre-trained model state trained on a subset of COCO2014 which contains 6,000 images
• a=0.44, b=0.28 for iwt and dwt