Frequency-Guided Iterative Network for Image Steganography (FIS)

The implementation of "Frequency-Guided Iterative Network for Image Steganography".

Getting Started

Prerequisites

Environment

• Python >= 3.6
• PyTorch >= 1.10.2

Datasets

Download the datasets subsampled image datasets (same as LISO). It is the subset of Div2k, CelebA, and MS COCO. To use them, set your dataset path in the 'utils.py'.

Training

python main.py --bits 2 --dataset div2k

Evaluation

python main.py --eval --bits 2 --dataset div2k --load checkpoints/div2k/2_bits.steg

Running Example

This is the running example of my project.

Acknowledgements

• LISO
• CBAM
• TrajGRU
• Extensive Reading During Research