blob: c1b7e52ff6f512f0234586266d3b2c4c2a8c544d (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
# -*- coding: utf-8 -*-
import os
import cv2
from imwatermark import WatermarkDecoder
input_file = os.environ['image']
output_file_path = os.environ['out']
num_bits = int(os.environ['num_bits'])
method = os.environ['method']
bgr = cv2.imread(input_file)
decoder = WatermarkDecoder('bytes', num_bits)
watermark = decoder.decode(bgr, method)
message = watermark.decode('utf-8')
with open(output_file_path, 'w') as f:
f.write(message)
|
