在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文章
21675浏览量
601961 -
图像处理
+关注
关注
27文章
1282浏览量
56656 -
python
+关注
关注
56文章
4782浏览量
84468
原文标题:FPGA图像处理--高斯模糊(二)
文章出处:【微信号:FPGA开源工坊,微信公众号:FPGA开源工坊】欢迎添加关注!文章转载请注明出处。
发布评论请先 登录
相关推荐
评论