/
githubmirror
/
ColossalAI
Обзор
Документация
Войти
/
githubmirror
/
ColossalAI
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
examples/images/diffusion/scripts/tests/test_watermark.py
18 строк
358 B
Hongxin Liu
[misc] update pre-commit and run all files (#4752)
19 сен 2023, 09:20
Не верифицирован
19 сен 2023, 09:20
079bf3c
Код
Авторство
О чём код?
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)