From d31dd20355492465b6d471aceee335067217ba50 Mon Sep 17 00:00:00 2001
From: Xi Xu <i@xi-xu.me>
Date: Mon, 2 Sep 2024 16:43:04 +0800
Subject: [PATCH] Update 1.py

---
 3/1.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/3/1.py b/3/1.py
index bf86363..33a8a78 100644
--- a/3/1.py
+++ b/3/1.py
@@ -1,6 +1,6 @@
 import cv2 as cv
-import numpy as np
 import matplotlib.pyplot as plt
+import numpy as np
 
 plt.rcParams["font.family"] = ["SimSun"]
 src = cv.imread(r"img\peppers.bmp", cv.IMREAD_GRAYSCALE)
@@ -10,7 +10,7 @@ row, col = src.shape[:2]
 src1 = np.zeros((row, col), dtype=np.float32)
 for i in range(row):
     for j in range(col):
-        src1[i, j] = src[i, j] * ((-1) ** (i + j))
+        src1[i, j] = np.float32(src[i, j]) * ((-1) ** (i + j))
 dft1 = cv.dft(src1, flags=cv.DFT_COMPLEX_OUTPUT)
 magnitude0 = cv.magnitude(dft1[:, :, 0], dft1[:, :, 1])
 magnitude1 = 20 * np.log(1 + magnitude0)  # type: ignore