646b892680
Some checks failed
Periodic Merges (6h) / master → staging-nixos (push) Failing after 12m50s
Periodic Merges (6h) / master → staging-next (push) Failing after 12m54s
Periodic Merges (24h) / merge-base(master,staging) → haskell-updates (push) Failing after 11m54s
Periodic Merges (6h) / staging-next → staging (push) Failing after 12m13s
Periodic Merges (24h) / staging-next-25.05 → staging-25.05 (push) Failing after 13m24s
Periodic Merges (24h) / release-25.05 → staging-next-25.05 (push) Failing after 14m28s
22 lines
497 B
Python
22 lines
497 B
Python
# -*- coding: utf-8 -*-
|
|
|
|
import os
|
|
import cv2
|
|
from imwatermark import WatermarkEncoder
|
|
|
|
input_file_path = os.environ['image']
|
|
output_dir = os.environ['out']
|
|
message = os.environ['message']
|
|
method = os.environ['method']
|
|
|
|
os.mkdir(output_dir)
|
|
|
|
bgr = cv2.imread(input_file_path)
|
|
|
|
encoder = WatermarkEncoder()
|
|
encoder.set_watermark('bytes', message.encode('utf-8'))
|
|
bgr_encoded = encoder.encode(bgr, method)
|
|
|
|
output_file = os.path.join(output_dir, 'test_wm.png')
|
|
cv2.imwrite(output_file, bgr_encoded)
|