Skip to content

[codex] add neural watermark scaffold #12

[codex] add neural watermark scaffold

[codex] add neural watermark scaffold #12

Workflow file for this run

name: test all
on: [push, pull_request]
permissions:
contents: read
jobs:
build:
runs-on: ubuntu-latest
name: Test robust watermark engine
defaults:
run:
working-directory: blind_watermark
steps:
- uses: actions/checkout@v6
- uses: actions/setup-python@v6
with:
python-version: '3.10'
- name: Install dependencies
run: |
pip install -r requirements.txt
- name: Smoke test - import and roundtrip
shell: python {0}
env:
PYTHONPATH: .
run: |
from rwm_engine import embed_watermark, extract_watermark, check_dependencies
import numpy as np
dep = check_dependencies()
assert dep['ok'], f'Missing deps: {dep}'
img = np.random.randint(0, 256, (800, 800, 3), dtype=np.uint8)
embedded = embed_watermark(
img,
'hello world',
password=42,
quality='balanced',
)
assert embedded['ok'], f'Embed failed: {embedded}'
assert embedded['image'] is not None, f'Embed image missing: {embedded}'
assert embedded['engine_used'] in {'legacy', 'neural'}, embedded
extracted = extract_watermark(
embedded['image'],
password=42,
quality='balanced',
)
assert extracted['ok'], f'Extract failed: {extracted}'
assert extracted['wm'] == 'hello world', f'Roundtrip failed: {extracted}'
print('All tests passed.')