stable-diffusion-stability-ai/scripts/tests/test_watermark.py

18 lines
357 B
Python
Raw Normal View History

2022-11-24 00:22:28 +00:00
import cv2
import fire
from imwatermark import WatermarkDecoder
def testit(img_path):
bgr = cv2.imread(img_path)
decoder = WatermarkDecoder('bytes', 136)
watermark = decoder.decode(bgr, 'dwtDct')
try:
dec = watermark.decode('utf-8')
except:
dec = "null"
print(dec)
if __name__ == "__main__":
fire.Fire(testit)