diff options
Diffstat (limited to 'neural_style.py')
| -rw-r--r-- | neural_style.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/neural_style.py b/neural_style.py index b62de21..6dc2cf5 100644 --- a/neural_style.py +++ b/neural_style.py @@ -707,10 +707,10 @@ def get_content_image(content_img): # resize if > max size if h > w and h > mx: w = (float(mx) / float(h)) * w - img = cv2.resize(img, dsize=(int(w), mx), interpolation=cv2.INTER_CUBIC) + img = cv2.resize(img, dsize=(int(w), mx), interpolation=cv2.INTER_AREA) if w > mx: h = (float(mx) / float(w)) * h - img = cv2.resize(img, dsize=(mx, int(h)), interpolation=cv2.INTER_CUBIC) + img = cv2.resize(img, dsize=(mx, int(h)), interpolation=cv2.INTER_AREA) img = preprocess(img, vgg19_mean) return img @@ -721,7 +721,7 @@ def get_style_images(content_img): path = os.path.join(args.style_imgs_dir, style_fn) # bgr image img = cv2.imread(path, cv2.IMREAD_COLOR).astype(np.float32) - img = cv2.resize(img, dsize=(cw, ch)) + img = cv2.resize(img, dsize=(cw, ch), interpolation=cv2.INTER_AREA) img = preprocess(img, vgg19_mean) style_imgs.append(img) return style_imgs @@ -735,7 +735,7 @@ def get_noise_image(noise_ratio, content_img): def get_mask_image(mask_img, width, height): path = os.path.join(args.content_img_dir, mask_img) img = cv2.imread(path, cv2.IMREAD_GRAYSCALE) - img = cv2.resize(img, dsize=(width, height)).astype(np.float32) + img = cv2.resize(img, dsize=(width, height), interpolation=cv2.INTER_AREA).astype(np.float32) mx = np.amax(img) img /= mx return img @@ -836,4 +836,4 @@ def main(): else: render_single_image() if __name__ == '__main__': - main()
\ No newline at end of file + main() |