Я пытаюсь реализовать алгоритм маркировки подключенных компонентов в Vivado HLS. Вот ссылка на алгоритм, который я использую: двухпроходный алгоритм маркировки связанного компонента.
Я использую Окно 3х3 в Vivado HLS и отправить это окно в мою функцию алгоритма подключенного объекта. Эта функция возвращает один пиксель в соответствии с алгоритмом и сохраняет его в целевом изображении и добавляет последовательный пиксель по мере их поступления.
Но при обработке следующего окна оно игнорирует результат предыдущей операции, а алгоритм требует, чтобы оно сравнивалось с предыдущим выходным пикселем.
Мне нужно найти способ рассмотреть этот пиксель или внести изменения в само изображение назначения.
Вот мой код:
conn.cpp
#include "CONN.h"int label=50;
int min;
MY_PIXEL find_conn(MY_WINDOW *window)
{unsigned char west, north, northWest, northEast, south, east, southEast, southWest=0;MY_PIXEL pixel;
char i=0;
char j=0;
// pixel.val[0]=window->getval(i,j); //to make copy of original image.
west=window->getval(i+1,j);
northWest=window->getval(i,j);
north=window->getval(i,j+1);
northEast=window->getval(i,j+2);
if(window->getval(i+1,j+1)!=0){
min=9600;
if(west!=0||north!=0||northWest!=0||northEast!=0){
if(west<min && west!=0) min=west;
if(northWest<min && north!=0) min=northWest;
if(north<min && north!=0) min=north;
if(northEast!=0 && northEast!=0) min=northEast;
window->insert(min,i+1,j+1);
}
else
{
label= label+10;
window->insert(label,i+1,j+1);
}}pixel.val[0]=window->getval(i+1,j+1);return pixel;
}
void created_window(MY_IMAGE& src, MY_IMAGE& dst, int rows, int cols)
{
MY_BUFFER buff_A;
MY_WINDOW WINDOW_3x3;
for(int row = 0; row < rows+1; row++){
for(int col = 0; col < cols+1; col++){
#pragma HLS loop_flatten off
#pragma HLS dependence variable=&buff_A false
#pragma HLS PIPELINE II = 1
// Temp values are used to reduce the number of memory reads
unsigned char temp;
MY_PIXEL tempx;
//Line Buffer fill
if(col < cols){
buff_A.shift_down(col);
temp = buff_A.getval(0,col);
}
//There is an offset to accommodate the active pixel region
//There are only MAX_WIDTH and MAX_HEIGHT valid pixels in the image
if(col < cols && row < rows){
MY_PIXEL new_pix;
src >> new_pix;
tempx = new_pix;
buff_A.insert_bottom(tempx.val[0],col);
}
//Shift the processing window to make room for the new column
WINDOW_3x3.shift_right();
//The processing window only needs to store luminance values
//rgb2y function computes the luminance from the color pixel
if(col < cols){
WINDOW_3x3.insert(buff_A.getval(2,col),2,0);
WINDOW_3x3.insert(temp,1,0);
WINDOW_3x3.insert(tempx.val[0],0,0);
}
MY_PIXEL conn_obj;//The operator only works on the inner part of the image
//This design assumes there are no connected objects on the boundary of the imageconn_obj = find_conn(&WINDOW_3x3);//The output image is offset from the input to account for the line buffer
if(row > 0 && col > 0) {
dst << conn_obj;
}
}
}
}
void create_window(AXI_STREAM& video_in, AXI_STREAM& video_out, int rows, int cols)
{
//Create AXI streaming interfaces for the core
#pragma HLS INTERFACE axis port=video_in bundle=INPUT_STREAM
#pragma HLS INTERFACE axis port=video_out bundle=OUTPUT_STREAM
#pragma HLS INTERFACE s_axilite port=rows bundle=CONTROL_BUS offset=0x14
#pragma HLS INTERFACE s_axilite port=cols bundle=CONTROL_BUS offset=0x1C
#pragma HLS INTERFACE s_axilite port=return bundle=CONTROL_BUS
#pragma HLS INTERFACE ap_stable port=rows
#pragma HLS INTERFACE ap_stable port=colsMY_IMAGE img_0(rows, cols);
MY_IMAGE img_1(rows, cols);
#pragma HLS dataflow
hls::AXIvideo2Mat(video_in, img_0);
created_window(img_0, img_1, rows, cols);
hls::Mat2AXIvideo(img_0, video_out);
}
conn.h
#ifndef _TOP_H_
#define _TOP_H_
#include "hls_video.h"
#define MAX_WIDTH 320
#define MAX_HEIGHT 240
typedef hls::stream<ap_axiu<8,1,1,1> > AXI_STREAM;
typedef hls::Scalar<1, unsigned char> MY_PIXEL;
typedef hls::Mat<MAX_HEIGHT, MAX_WIDTH, HLS_8UC1> MY_IMAGE;
typedef hls::Window<3, 3, unsigned char> MY_WINDOW;
typedef hls::LineBuffer<3, MAX_WIDTH, unsigned char> MY_BUFFER;
void create_window(AXI_STREAM& INPUT_STREAM, AXI_STREAM& OUTPUT_STREAM, int rows, int cols);
#endif
В соответствии с просьбой здесь мой main.c
#include "CONN.h"#include <hls_video.h>
#include <hls_opencv.h>
#define INPUT_FILE "D:/TRACKING PRADNAYA/test_image/test_conn.jpg"#define OUTPUT_FILE "D:/TRACKING PRADNAYA/test_image/test_conn_out.jpg"
int main(){
cv::Mat inputMat(MAX_HEIGHT, MAX_WIDTH, CV_8UC1);
cv::Mat outputMat(MAX_HEIGHT, MAX_WIDTH, CV_8UC1);
inputMat = cv::imread(INPUT_FILE, CV_LOAD_IMAGE_GRAYSCALE);AXI_STREAM inStream_1;
AXI_STREAM outStream_conn;cvMat2AXIvideo(inputMat, inStream_1);create_window(inStream_1, outStream_conn, inputMat.rows, inputMat.cols);
AXIvideo2cvMat(outStream_conn, outputMat);cv::imwrite(OUTPUT_FILE, outputMat );return 0;
}
Задача ещё не решена.
Других решений пока нет …