/
githubmirror
/
stable-diffusion
Обзор
Документация
Войти
/
githubmirror
/
stable-diffusion
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
scripts/tests/test_watermark.py
18 строк
357 B
Patrick Esser
Release under CreativeML Open RAIL M License
22 авг 2022, 19:57
22 авг 2022, 19:57
69ae4b3
Код
Авторство
О чём код?
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)