hggjcqwe/RANDMark_demo

🏷️ RANDmark Demo Models

RANDmark 数字水印系统的预训练模型，用于 Gradio Demo 演示。

📦 模型文件

文件	描述	大小
encoder_best.pth	编码器模型	~9MB
decoder_best.pth	解码器模型	~8MB
seg_92.pth	分割模型 (几何校正)	~15MB

🚀 使用方法

from huggingface_hub import hf_hub_download
import torch

# 下载模型
encoder_path = hf_hub_download(repo_id="ghgjcqwe/RANDmark-demo", filename="encoder_best.pth")
decoder_path = hf_hub_download(repo_id="ghgjcqwe/RANDmark-demo", filename="decoder_best.pth")

# 加载模型
encoder = EncoderDecoder(...)
encoder.encoder.load_state_dict(torch.load(encoder_path))
encoder.decoder.load_state_dict(torch.load(decoder_path))

🔗 相关链接

• Demo Space
• GitHub 仓库