mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-05-13 10:57:20 +00:00
ca6e75bc28
* Add handson to ColossalAI. * Change names of handsons and edit sequence parallel example. * Edit wrong folder name * resolve conflict * delete readme
18 lines
357 B
Python
18 lines
357 B
Python
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) |