- Metaが発表した「Watermark Anything with Localized Messages」論文の公式実装
- 画像に局所化されたウォーターマークを埋め込める方法を提供
- 複数のウォーターマークを埋め込める
- 要件: Python 3.10.14、PyTorch 2.5.1、CUDA 12.4、Torchvision 0.20.1 でテスト済み。
- Inference
notebooks/inference.ipynb で推論スクリプトと可視化を確認できる
- 画像にウォーターマークを埋め込み、検出・デコードする方法を説明
wam.scaling_w 引数を調整することで、ウォーターマークの imperceptibility/robustness(目立ちにくさ/堅牢性)のバランスを調整できる
- 単一ウォーターマーク
- 32ビットメッセージを画像に埋め込む例を提供
- 画像の一部にのみウォーターマークを埋め込めるランダムマスクを生成
- 複数ウォーターマーク
- 複数の32ビットメッセージを画像に埋め込む例を提供
- DBSCANアルゴリズムを使い、クラスタリングによってウォーターマークを検出
- 学習
- 事前学習: 堅牢性のための事前学習方法を説明。
- ファインチューニング: 複数ウォーターマークと目立ちにくさを保証するためのモデルのファインチューニング方法を説明。
- ライセンス
- モデルは CC-BY-NC ライセンスで提供される。
- 参考
- AudioSeal、Segment Anything プロジェクトと関連がある。
1件のコメント
Hacker Newsの意見