tjh.dev
1# -*- coding: utf-8 -*-
2
3import os
4import cv2
5from imwatermark import WatermarkEncoder
6
7input_file_path = os.environ['image']
8output_dir = os.environ['out']
9message = os.environ['message']
10method = os.environ['method']
11
12os.mkdir(output_dir)
13
14bgr = cv2.imread(input_file_path)
15
16encoder = WatermarkEncoder()
17encoder.set_watermark('bytes', message.encode('utf-8'))
18bgr_encoded = encoder.encode(bgr, method)
19
20output_file = os.path.join(output_dir, 'test_wm.png')
21cv2.imwrite(output_file, bgr_encoded)