在FPGA圖像處理--高斯模糊(一)中介紹了怎么使用Python生成高斯模糊需要使用的高斯核,在這個文章中就介紹一下怎么在FPGA中完成高斯模糊。
首先我們使用Python生成一個高斯模糊的參考模型,代碼如下:
import cv2 as cv
import numpy as np
img_path = "./sim/img/img.png"
img = cv.imread(img_path)
img_gray = cv.cvtColor(img, cv.COLOR_RGB2GRAY)
h, w = img_gray.shape
# kernel_1d = cv.getGaussianKernel(3, 0.8, ktype=cv.CV_32F)
# kernel_2d = kernel_1d * kernel_1d.T
# kernel_2d = kernel_2d * (1 / kernel_2d[0][0])
# print(kernel_2d)
gauss_kernel = [[1, 2, 1],
[2, 4, 2],
[1, 2, 1]]
img_padding = np.zeros((h + 2, w + 2), np.uint8)
img_padding[1:h + 1, 1:w + 1] = img_gray
img_padding[0:1, 1:w + 1] = img_gray[0:1, :]
img_padding[h + 1:h + 2, 1:w + 1] = img_gray[h - 1:h, :]
img_padding[:, 0:1] = img_padding[:, 1:2]
img_padding[:, w + 1:w + 2] = img_padding[:, w:w + 1]
# gauss_img = cv.GaussianBlur(img_gray, (3, 3), 0.8)
gauss_img = np.zeros((h, w), np.uint8)
for i in range(1, h + 1):
for j in range(1, w + 1):
gauss_img[i - 1][j - 1] = (img_padding[i - 1][j - 1] * gauss_kernel[0][0] + img_padding[i - 1][j] *
gauss_kernel[0][1] + img_padding[i - 1][j + 1] * gauss_kernel[0][2] + img_padding[i][
j - 1] * gauss_kernel[1][0] + img_padding[i][j] * gauss_kernel[1][1] +
img_padding[i][j + 1] * gauss_kernel[1][2] + img_padding[i + 1][j - 1] *
gauss_kernel[2][0] + img_padding[i + 1][j] * gauss_kernel[2][1] + img_padding[i + 1][
j + 1] * gauss_kernel[2][2]) / 16
def save_file(img, path):
fp = open(path,'w')
for i in range(img.shape[0]):
for j in range(img.shape[1]):
if i == img.shape[0] - 1 and j == img.shape[1] - 1:
fp.write(str(img[i][j]))
else:
fp.write(str(img[i][j]) + "
")
fp.close
def img2txt(img, dst):
f = open(dst,"w")
for i in range(img.shape[0]):
for j in range(img.shape[1]):
if i == img.shape[0] - 1 and j == img.shape[1] - 1:
f.write(str(img[i][j][0]) + " " + str(img[i][j][1]) + " " + str(img[i][j][2]))
else:
f.write(str(img[i][j][0]) + " " + str(img[i][j][1]) + " " + str(img[i][j][2]) + "
")
f.close()
file_path = "./sim/test.txt"
file_path_ref = "./sim/ref.txt"
img2txt(img, file_path)
save_file(gauss_img, file_path_ref)
cv.imshow("lena", img_gray)
cv.imshow("lena_gauss_img", gauss_img)
cv.waitKey()
cv.destroyAllWindows()
在上述代碼中完成了讀取一張圖片,然后進行轉灰度和進行高斯模糊的操作,并將原圖和高斯模糊后圖像數據存到了兩個txt中,用于我們仿真。
在FPGA中完成高斯模糊比較簡單,使用以下代碼完成高斯矩陣的乘加。
仿真代碼:
首先使用一個模塊來讀取之前用python生成的原圖數據,用于高斯模糊的激勵。
然后例化我們寫的高斯模糊模塊
最后將仿真結果保存起來,并且在仿真的時候對dut輸出的結果和參考模型進行比對,如果出現錯誤就停止仿真。
仿真結果如下:
在仿真的時候會實時打印DUT和參考模型的結果是否比對成功。因為設置了DUT和參考模型的結果之間的閾值為5,所以當兩者差值在5以內時都會打印sim success。
仿真對比如下:
-
FPGA
+關注
關注
1626文章
21666瀏覽量
601830 -
圖像處理
+關注
關注
27文章
1281瀏覽量
56638 -
python
+關注
關注
56文章
4782瀏覽量
84453
原文標題:FPGA圖像處理--高斯模糊(二)
文章出處:【微信號:FPGA開源工坊,微信公眾號:FPGA開源工坊】歡迎添加關注!文章轉載請注明出處。
發布評論請先 登錄
相關推薦
評論