Add files using upload-large-folder tool

README.md

@@ -11,6 +11,8 @@ tags:
   - checkpoint
 ---
 
+> [!WARNING] we do not have a full checkpoint conversion validation, if you encounter pipeline loading failure and unsidered output, please contact me via bili_sakura@zju.edu.cn
+
 # BiliSakura/StegoGAN-ckpt
 
 Packaged StegoGAN checkpoints from `models/raw/StegoGAN`.
@@ -38,6 +40,29 @@ StegoGAN-ckpt/
       diffusion_pytorch_model.safetensors
 ```
 
+## Usage (Inference)
+
+```python
+import numpy as np
+import torch
+from PIL import Image
+
+from src.pipelines.stegogan import StegoGANPipeline
+
+pipe = StegoGANPipeline.from_pretrained(
+    "/path/to/StegoGAN-ckpt/google_mismatch_0.25",
+    device="cuda",
+)
+
+img = Image.open("/path/to/input.png").convert("RGB")
+x = torch.from_numpy(np.array(img)).float().permute(2, 0, 1).unsqueeze(0) / 255.0
+x = x * 2 - 1
+x = x.to("cuda")
+
+out = pipe(source=x, direction="a2b", output_type="pil")
+out.images[0].save("stegogan_output.png")
+```
+
 ## Notes
 
 - `generator_A/` and `generator_B/` safetensors exports are included for each variant.