FPGA圖像處理基礎(chǔ)----實(shí)現(xiàn)緩存卷積窗口

像素行與像素窗口

一幅圖像是由一個(gè)個(gè)像素點(diǎn)構(gòu)成的，對(duì)于一幅480*272大小的圖片來說，其寬度是480，高度是272。在使用FPGA進(jìn)行圖像處理時(shí)，最關(guān)鍵的就是使用FPGA內(nèi)部的存儲(chǔ)資源對(duì)像素行進(jìn)行緩存與變換。由于在圖像處理過程中，經(jīng)常會(huì)使用到卷積，因此需要對(duì)圖像進(jìn)行開窗，然后將開窗得到的局部圖像與卷積核進(jìn)行卷積，從而完成處理。
??圖像數(shù)據(jù)一般按照一定的格式和時(shí)序進(jìn)行傳輸，在我進(jìn)行實(shí)驗(yàn)的時(shí)候，處理圖像時(shí)，讓其以VGA的時(shí)序來進(jìn)行工作，這樣能夠?yàn)槲姨幚硇芯彺嫣峁┍憷?/p>

基于FIFO的行緩存結(jié)構(gòu)

??在FPGA中對(duì)圖像的一行數(shù)據(jù)進(jìn)行緩存時(shí)，可以采用FIFO這一結(jié)構(gòu)，如上圖所示，新一行圖像數(shù)據(jù)流入到FIFO1中，F(xiàn)IFO1中會(huì)對(duì)圖像數(shù)據(jù)進(jìn)行緩存，當(dāng)FIFO1中緩存有一行圖像數(shù)據(jù)時(shí)，在下一行圖像數(shù)據(jù)來臨的時(shí)候，將FIFO1中緩存的圖像數(shù)據(jù)讀出，并傳遞給下一個(gè)FIFO，于此同時(shí)，將新一行的圖像數(shù)據(jù)緩存到FIFO1中，這樣就能完成多行圖像的緩存。
??若要緩存多行圖像，下面的菊花鏈?zhǔn)降慕Y(jié)果更能夠直觀地表現(xiàn)圖像數(shù)據(jù)地流向。

??新輸入地圖像緩存到FIFO1當(dāng)中，當(dāng)FIFO中緩存有一行數(shù)據(jù)的時(shí)候，下一個(gè)輸入像素來臨的時(shí)候，會(huì)將數(shù)據(jù)從本FIFO中讀出，并給到下一個(gè)FIFO，來形成類似于一個(gè)流水線的結(jié)構(gòu)。
??上面的圖中，就是實(shí)現(xiàn)一個(gè)5X5大小的窗口的一個(gè)結(jié)構(gòu)圖。

代碼設(shè)計(jì)

實(shí)現(xiàn)一個(gè)可以生成任意尺寸大小的開窗的模塊，需要注意參數(shù)的使用，可以通過調(diào)節(jié)KSZ來調(diào)整窗口的大小。最終將窗口中的圖像像素，轉(zhuǎn)換成一個(gè)一維的數(shù)據(jù)輸出給到下一個(gè)模塊。
??在設(shè)計(jì)的時(shí)候，對(duì)于FIFO要選擇精準(zhǔn)計(jì)數(shù)模式，這樣才能讓流水正常工作起來。
??在代碼中通過generate語句來實(shí)現(xiàn)多個(gè)line_buffer的例化，line buffer的個(gè)數(shù)可以根據(jù)卷積窗口的大小來選擇，例如3X3大小的卷積窗口需要緩存兩行，5X5大小的卷積窗口需要緩存4行，可以通過設(shè)置參數(shù)來選擇要例化多少個(gè)line_buffer。

時(shí)序設(shè)計(jì)

??在設(shè)計(jì)FIFO的菊花鏈結(jié)構(gòu)時(shí)，需要根據(jù)當(dāng)前FIFO中存儲(chǔ)的數(shù)據(jù)個(gè)數(shù)來判斷，這時(shí)候使用到精準(zhǔn)計(jì)數(shù)模式，可以反應(yīng)FIFO中的存儲(chǔ)的數(shù)據(jù)。當(dāng)FIFO中存儲(chǔ)有一行數(shù)據(jù)的時(shí)候，使能pop_en信號(hào)，表示當(dāng)前可以將數(shù)據(jù)從FIFO中讀出。
??在將數(shù)據(jù)寫入到FIFO中的時(shí)候，需要對(duì)數(shù)據(jù)進(jìn)行擴(kuò)充，也即需要對(duì)輸入的圖像的邊界補(bǔ)充數(shù)據(jù)，因?yàn)檫M(jìn)行卷積之后的圖像將會(huì)比原始圖像數(shù)據(jù)尺寸減少，因此在形成卷積窗口時(shí)，將圖像擴(kuò)充，能夠讓圖像處理完成之后，保持原來的尺寸，只是會(huì)在邊界出現(xiàn)黑邊。
??win_buf這個(gè)模塊的最終輸出，就是一個(gè)矩陣內(nèi)的所有像素，組成一個(gè)信號(hào)輸出到外部，供進(jìn)行卷積的處理。

/*============================================
#
# Author: Wcc - 1530604142@qq.com
#
# QQ : 1530604142
#
# Last modified: 2020-07-08 20:02
#
# Filename: win_buffer.v
#
# Description: 
#
============================================*/
`timescale 1ns / 1ps
module win_buf #(
//==========================================
//parameter define
//==========================================
parameterKSZ = 3,//卷積核大小
parameterIMG_WIDTH = 128,//圖像寬度,每個(gè)Line_buffer需要緩存的數(shù)據(jù)量
parameterIMG_HEIGHT= 128 //圖像高度
)(
input wire clk ,
inputwirerst ,
inputwirepi_hs,
inputwirepi_vs,
inputwirepi_de,
inputwirepi_dv,
inputwire[7: 0]pi_data,
//輸出圖像新有效信號(hào)
output wirepo_dv,
inputwirepo_hs,
inputwirepo_vs,
inputwirepo_de,
//輸出3X3圖像矩陣
outputwire[8*KSZ*KSZ - 1: 0]image_matrix
    );
//==========================================
//internal signals
//==========================================
reg frame_flag;//當(dāng)前處于一幀圖像所在位置

//==========================================
//linebuffer 緩存數(shù)據(jù)
//==========================================
//KSZ * KSZ 大小的卷積窗口,需要緩存(KSZ - 1行)
wire [7:0]wr_buf_data[KSZ-2: 0];//寫入每個(gè)buffer的數(shù)據(jù)
wire wr_buf_en  [KSZ-2: 0];//寫每個(gè)buffer的使能
wire full [KSZ-2: 0];//每個(gè)buffer的空滿信號(hào)
wire empty [KSZ-2: 0];
wire [11:0]rd_data_cnt [KSZ-2: 0];//每個(gè)bufeer的內(nèi)存數(shù)據(jù)個(gè)數(shù)
wirerd_buf_en[KSZ-2: 0];//讀出每個(gè)buffer的使能信號(hào)
wire [7:0]rd_buf_data[KSZ-2: 0];//讀出每個(gè)buffer的數(shù)據(jù)
reg pop_en [KSZ-2: 0];//每個(gè)buffer可以讀出數(shù)據(jù)信號(hào)

reg pi_dv_dd [KSZ-2 : 0];//輸入有效數(shù)據(jù)延時(shí)
wireline_vld;//行圖像數(shù)據(jù)有效信號(hào)
wire [7:0]in_line_data;
reg [7:0]pi_data_dd [KSZ-2 : 0];
reg [1:0]pi_vs_dd;

reg [KSZ-2 : 0]po_dv_dd;
reg [KSZ-2 : 0]po_de_dd;
reg [KSZ-2 : 0]po_hs_dd;
reg [KSZ-2 : 0]po_vs_dd;

reg [0 : 0]po_dv_r;
reg [0 : 0]po_de_r;
reg [0 : 0]po_hs_r;
reg [0 : 0]po_vs_r;

reg [12:0]cnt_col ;//行列計(jì)數(shù)器
wireadd_cnt_col;
wireend_cnt_col;
reg [12:0]cnt_row ;
wireadd_cnt_row;
wireend_cnt_row ;

localparamMATRIX_SIZE = KSZ * KSZ;

//輸出列數(shù)據(jù)延時(shí)，形成矩陣
reg[7:0]matrix_data[MATRIX_SIZE - 1: 0];
reg [8*KSZ*KSZ - 1: 0]image_matrix_r;


assign  image_matrix = image_matrix_r;
assign po_dv = po_dv_r;
assign po_hs = po_hs_r;
assign po_vs = po_vs_r;
assign po_de = po_de_r;


//----------------pi_vs_dd------------------
always @(posedge clk) begin
if (rst==1'b1) begin
pi_vs_dd <= 'd0;
end
else begin
pi_vs_dd <= {pi_vs_dd[0], pi_vs};
end
end

//----------------frame_flag------------------
always @(posedge clk) begin
if (rst==1'b1) begin
frame_flag <= 1'b0;
end
//檢測(cè)到上升沿，結(jié)束上一幀
else if (pi_vs_dd[0] == 1'b1 && pi_vs_dd[1] == 1'b0) begin
frame_flag <= 1'b0;
end
//檢測(cè)到下降沿，開始本幀
else if(pi_vs_dd[0] == 1'b0 && pi_vs_dd[1] == 1'b1) begin
frame_flag <= 1'b1;
end
end

//==========================================
//pi_dv_dd
//進(jìn)行邊界的擴(kuò)充，以3X3的窗口為例,需要在圖像的
//外圍各邊添加一行一列的的空白像素
//若是5X5的窗口，需要在圖像的外圍各邊，添加2行2列
//==========================================
generate
genvar i;
//對(duì)于KSZ*KSZ的卷積核，每一行需要延時(shí)(KSZ-1)拍
for (i = 0; i < KSZ-1; i = i + 1)
begin:Expand_the_boundary_dv
if (i == 0) begin
always @(posedge clk) begin
if (rst==1'b1) begin
pi_dv_dd[i] <= 1'b0;
pi_data_dd[i] <= 'd0;
end
else begin
pi_dv_dd[i] <= pi_dv;
pi_data_dd[i] <= pi_data;
end
end
end
else begin
always @(posedge clk) begin
if (rst==1'b1) begin
pi_dv_dd[i] <= 1'b0;
pi_data_dd[i] <= 'd0;
end
else begin
pi_dv_dd[i] <= pi_dv_dd[i - 1];
pi_data_dd[i] <= pi_data_dd[i - 1];
end
end
end
end
endgenerate

assign line_vld = pi_dv_dd[KSZ-2] | pi_dv;
assign in_line_data = pi_data_dd[(KSZ>>1) - 1];

//==========================================
//行列計(jì)數(shù)器
//==========================================
//----------------cnt_col------------------
always @(posedge clk) begin
if (rst == 1'b1) begin
cnt_col <= 'd0;
end
else if (add_cnt_col) begin
if(end_cnt_col)
cnt_col <= 'd0;
else
cnt_col <= cnt_col + 1'b1;
end
else begin
cnt_col <= 'd0;
end
end

assign add_cnt_col = frame_flag == 1'b1 && line_vld == 1'b1;
assign end_cnt_col = add_cnt_col &&cnt_col == (IMG_WIDTH + KSZ-1) - 1;

//----------------cnt_row------------------
always @(posedge clk) begin
if (rst == 1'b1) begin
cnt_row <= 'd0;
end
else if (add_cnt_row) begin
if(end_cnt_row)
cnt_row <= 'd0;
else
cnt_row <= cnt_row + 1'b1;
end
end

assign add_cnt_row = end_cnt_col == 1'b1 ;
assign end_cnt_row = add_cnt_row &&cnt_row == IMG_HEIGHT - 1;

//==========================================
//輸入到line_buffer,構(gòu)成對(duì)齊的列
//通過例化FIFO的方式,來完成行緩存
//當(dāng)FIFO緩存有一行數(shù)據(jù)時(shí)，將數(shù)據(jù)讀出，并填充到下一個(gè)FIFO中
//==========================================
generate
genvar j;
for (j = 0; j < KSZ - 1; j = j + 1)
begin:multi_line_buffer
//第一個(gè) line_buffer
if (j == 0) begin : first_buffer
//寫入第一個(gè)line_buffer的數(shù)據(jù)是從外部輸入的數(shù)據(jù)
assign wr_buf_data[j] = in_line_data;
assign wr_buf_en[j] = line_vld;
end
//其他line_buffer
else begin : other_buffer
//寫入其他line_buffer的數(shù)據(jù)是上一個(gè)line_buffer中輸出的數(shù)據(jù)
assign wr_buf_en[j] = rd_buf_en[j - 1] ;
assign wr_buf_data[j] = rd_buf_data[j - 1] ;
end

//----------------rd_buf_en------------------
//從buffer中讀出數(shù)據(jù)
//pop_en是當(dāng)前FIFO中已經(jīng)緩存了一行的圖像數(shù)據(jù)的指示信號(hào)
//wr_buf_en是當(dāng)前新一行寫入的有效數(shù)據(jù)指示信號(hào)
assign rd_buf_en[j] = pop_en[j] & wr_buf_en[j];

always @(posedge clk) begin
if (rst==1'b1) begin
pop_en[j] <= 0;
end
//當(dāng)前不處于圖像有效數(shù)據(jù)區(qū)域
else if (frame_flag == 1'b0) begin
pop_en[j] <= 1'b0;
end
//當(dāng)buffer中緩存有一行圖像數(shù)據(jù)時(shí)(擴(kuò)充的圖像行)
else if (rd_data_cnt[j] >= IMG_WIDTH+2) begin
pop_en[j] <= 1'b1;
end
end

line_buf line_buffer (
  .wr_clk(clk),                // input wire wr_clk
  .rd_clk(clk),                // input wire rd_clk
  .din(wr_buf_data[j]),        // input wire [7 : 0] din
  .wr_en(wr_buf_en[j]),        // input wire wr_en
  .rd_en(rd_buf_en[j]),                  // input wire rd_en
  .dout(rd_buf_data[j]),                    // output wire [7 : 0] dout
  .full(full[j]),                    // output wire full
  .empty(empty[j]),                  // output wire empty
  .rd_data_count(rd_data_cnt[j])  // output wire [11 : 0] rd_data_count
);
end
endgenerate

//==========================================
//得到矩陣中的每一行的第一個(gè)數(shù)據(jù)
//==========================================
generate
genvar k;
for (k = 0; k < KSZ; k = k + 1)
begin:matrix_data_first_col
if (k == KSZ -1) begin
//最后一行數(shù)據(jù)為剛輸入的數(shù)據(jù)
always @(*) begin
matrix_data[KSZ * k] = in_line_data;
end
end
else begin
//從buffer中讀取出來的圖像書籍
//以3X3為例，buffer是菊花鏈的結(jié)果
//最開始輸入的數(shù)據(jù)，在最后一個(gè)buffer中被讀出
//最后輸入的數(shù)據(jù)，在第一個(gè)buffer中被讀出
always @(*) begin
 matrix_data[KSZ * k] = rd_buf_data[(KSZ -2) - k];
end
end
end
endgenerate
//==========================================
//延時(shí)得到矩陣數(shù)據(jù)
//以3X3為例
//matrix[0]~[2] ==>p13,p12,p11
//matrix[3]~[5] ==>p23,p22,p21
//matrix[6]~[8] ==>p33,p32,p31
//KSZ-1 clk
//==========================================
generate
genvar r,c;
for (r = 0; r < KSZ; r = r + 1)
begin:row
for (c = 1; c < KSZ; c = c + 1)
begin:col
always @(posedge clk) begin
if (rst==1'b1) begin
matrix_data[r*KSZ + c] <= 'd0;
end
else begin
matrix_data[r*KSZ + c] <= matrix_data[r*KSZ + (c-1)];
end
end
end
end
endgenerate

//==========================================
//輸出圖像矩陣
//以3X3為例
//image_matrix [7:0] ==> p13
//image_matrix [15:8] ==> p12
//image_matrix [23:16] ==> p11
//1 clk
//==========================================
generate
genvar idx;
for (idx = 0; idx < KSZ*KSZ; idx = idx + 1)
begin:out_put_data
always @(posedge clk) begin
if (rst==1'b1) begin
po_dv_r <= 1'b0;
po_de_r <= 1'b0;
po_hs_r <= 1'b0;
po_vs_r <= 1'b0;
image_matrix_r[8*(idx+1) -1 : 8*(idx)] <= 'd0;
end
else begin
po_dv_r <= po_dv_dd[KSZ-2] & line_vld;
po_de_r <= po_de_dd[KSZ-2] & line_vld;
po_hs_r <= po_hs_dd[KSZ-2] ;
po_vs_r <= po_vs_dd[KSZ-2] ;
if (po_dv_dd[KSZ-2] & line_vld == 1'b1) begin
image_matrix_r[8*(idx+1) -1 : 8*(idx)] <= matrix_data[idx];
end
else begin
image_matrix_r[8*(idx+1) -1 : 8*(idx)] <= 'd0;
end
end
end 
end
endgenerate
//==========================================
//(KSZ-1) clk
//==========================================

always @(posedge clk) begin
if (rst==1'b1) begin
po_vs_dd <= 'd0;
po_hs_dd <= 'd0;
po_de_dd <= 'd0;
po_dv_dd <= 'd0;
end
else begin
po_vs_dd <= {po_vs_dd[KSZ-3:0], pi_vs};
po_hs_dd <= {po_hs_dd[KSZ-3:0], pi_hs};
po_de_dd <= {po_de_dd[KSZ-3:0], pi_de};
po_dv_dd <= {po_dv_dd[KSZ-3:0], pi_dv};
end
end
endmodule

仿真驗(yàn)證

3X3開窗

輸入的第三行數(shù)據(jù)的前三個(gè)數(shù)據(jù)是：0x00，0x78，0x7c
輸入的第二行數(shù)據(jù)的前三個(gè)數(shù)據(jù)是：0x00，0x7d，0x7d
輸入的第一行數(shù)據(jù)的前三個(gè)數(shù)據(jù)是：0x00，0x7e，0x7f
輸出的第一個(gè)矩陣的值是：0x0078_7c00_7d7d_007e_7f
輸入行數(shù)據(jù)第一個(gè)數(shù)據(jù)是0x00這是因?yàn)閿U(kuò)充了邊界的原因。
??可以看到，設(shè)置KSZ為3,可以得到一個(gè)位寬為72bit的輸出數(shù)據(jù)，該數(shù)據(jù)包含了一個(gè)窗口中的9個(gè)數(shù)據(jù)。

5X5開窗

設(shè)置開窗大小為5x5之后，也可以看到輸出信號(hào)的位寬變?yōu)榱?*25=200bit，也就是一個(gè)5X5大小的矩陣中的數(shù)據(jù)。

輸入的第5行數(shù)據(jù)的前5個(gè)數(shù)據(jù)是：0x00，0x00，0x7e，0x7c，0x7f
輸入的第4行數(shù)據(jù)的前5個(gè)數(shù)據(jù)是：0x00，0x00，0x7e，0x7e，0x7e，
輸入的第3行數(shù)據(jù)的前5個(gè)數(shù)據(jù)是：0x00，0x00，0x78，0x7c，0x7c
輸入的第2行數(shù)據(jù)的前5個(gè)數(shù)據(jù)是：0x00，0x00，0x7d，0x7d，0x7a
輸入的第1行數(shù)據(jù)的前5個(gè)數(shù)據(jù)是：0x00，0x00，0x7e，0x7f，0x7d
從輸出結(jié)果看，輸出的矩陣數(shù)據(jù)，剛好是這5行的前5數(shù)據(jù)，并且前兩個(gè)數(shù)據(jù)是0x00，這是因?yàn)樵诿恳恍星懊嫜a(bǔ)充了兩個(gè)0的原因。
??經(jīng)過測(cè)試，這種開窗算子是能夠完成任意此村的開窗的。

實(shí)際應(yīng)用

在實(shí)際應(yīng)用中，我也將這個(gè)模塊正確地使用上了，完成了一個(gè)3x3的sobel算子和5x5的均值濾波。

原始圖像

3x3 Sobel

原文鏈接：

https://tencentcloud.csdn.net/678a0adeedd0904849a65d80.html