Views
No views yet
saved_model format, designed to detect images manipulated through steganographic techniques. The model uses a Transformer-based architecture and is optimized for grayscale image inputs of size 256x256.(256, 256, 1)stego vs cover)saved_modelpip install tensorflow huggingface_hub1from huggingface_hub import snapshot_download
2import tensorflow as tf
3
4# Download the model from Hugging Face Hub
5model_path = snapshot_download(repo_id="MarilineDelgado/capsteg")
6
7# Load the model in SavedModel format
8model = tf.keras.models.load_model(model_path)1import numpy as np
2
3# Test image (1, 256, 256, 1) - grayscale
4image = np.random.rand(1, 256, 256, 1).astype(np.float32)
5
6# Make prediction
7prediction = model.predict(image)
8
9print("Prediction:", prediction)