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