From 7a628e4d4e4e962fbf04090504e232f0ecd43369 Mon Sep 17 00:00:00 2001 From: gaomingqi Date: Wed, 26 Apr 2023 17:06:18 +0800 Subject: [PATCH] update README and base_inpainter (fix resizing) --- .gitignore | 1 + README.md | 2 +- inpainter/base_inpainter.py | 7 +++---- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/.gitignore b/.gitignore index bb8c488..9566859 100644 --- a/.gitignore +++ b/.gitignore @@ -9,3 +9,4 @@ debug_images/ *.npy images/ test_sample/ +doc/ diff --git a/README.md b/README.md index 018f4d8..3f9016e 100644 --- a/README.md +++ b/README.md @@ -9,7 +9,7 @@ - + diff --git a/inpainter/base_inpainter.py b/inpainter/base_inpainter.py index 35d6d4e..68af998 100644 --- a/inpainter/base_inpainter.py +++ b/inpainter/base_inpainter.py @@ -64,6 +64,7 @@ class BaseInpainter: masks = np.stack([cv2.dilate(mask, kernel) for mask in masks], 0) T, H, W = masks.shape + masks = np.expand_dims(masks, axis=3) # expand to T, H, W, 1 # size: (w, h) if ratio == 1: size = None @@ -74,10 +75,8 @@ class BaseInpainter: if min(size) < 50: ratio = 50. / min(H, W) size = [int(W*ratio), int(H*ratio)] - - masks = np.expand_dims(masks, axis=3) # expand to T, H, W, 1 - binary_masks = resize_masks(masks, tuple(size)) - frames = resize_frames(frames, tuple(size)) # T, H, W, 3 + binary_masks = resize_masks(masks, tuple(size)) + frames = resize_frames(frames, tuple(size)) # T, H, W, 3 # frames and binary_masks are numpy arrays h, w = frames.shape[1:3]