opensource/DeOldify
3
0
Fork 0
mirror of https://github.com/jantic/DeOldify.git synced 2026-09-01 15:36:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

oops

Browse Source
This commit is contained in:
Jason Antic
2020-02-20 09:04:15 -08:00
parent 48efb2c46e
commit 51991de5e6
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
deoldify/visualize.py
+3 -3
View File
@@ -27,12 +27,12 @@ def get_watermarked(pil_image: Image) -> Image:
(h, w) = image.shape[:2]
image = np.dstack([image, np.ones((h, w), dtype="uint8") * 255])
pct = 0.05
(fwH, fwW) = full_watermark.shape[:2]
wH = int(pct * h)
wW = int((pct * h / fwH) * fwW)
full_watermark = cv2.imread(
'./resource_images/watermark.png', cv2.IMREAD_UNCHANGED
)
(fwH, fwW) = full_watermark.shape[:2]
wH = int(pct * h)
wW = int((pct * h / fwH) * fwW)
watermark = cv2.resize(full_watermark, (wH, wW), interpolation=cv2.INTER_AREA)
overlay = np.zeros((h, w, 4), dtype="uint8")
(wH, wW) = watermark.shape[:2]