【论文复现】CRoSS：Diffusion Model Makes Controllable, Robust and Secure Image Steganography

• 论文链接：CRoSS
• 开源代码：yujiwen/CRoSS

1. 环境配置

demo脚本中用到了cv2库，安装命令：

pip install opencv-python==4.5.5.64

pytoch安装命令：

# CUDA 12.1 conda install pytorch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 pytorch-cuda=12.1 -c pytorch -c nvidia

虚拟环境依赖：

pytho==3.9 torch==2.1.0 numpy==1.23.5 opencv-python==4.5.5.64 transformers==4.33.2 diffusers==0.21.2 huggingface-hub==0.17.2 safetensors==0.3.3

2. 运行结果

运行命令：

python demo.