/
githubmirror
/
stablediffusion
Обзор
Документация
Войти
/
githubmirror
/
stablediffusion
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
scripts/tests/test_watermark.py
18 строк
357 B
Robin Rombach
release more models
24 ноя 2022, 03:22
24 ноя 2022, 03:22
ca86da3
Код
Авторство
О чём код?
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)