S04-CH04 Skin_Dection仿真实验

#include "top.h"

#include <string.h>

void hls::hls_skin_dection(RGB_IMAGE& src, RGB_IMAGE& dst,int rows, int cols,

int y_lower,int y_upper,int cb_lower,int cb_upper,int cr_lower,int cr_upper)

{

LOOp_ROWS:for(int row = 0; row < rows ; row++)

{

LOOp_COLS:for(int col = 0; col < cols; col++)

{

     //变量定义

RGB_PIXEL src_data;

     RGB_PIXEL pix;

     RGB_PIXEL dst_data;

     bool skin_region;

        if(row < rows && col < cols) {

         src >> src_data;

        }

        //获取RGB通道数据

        uchar B = src_data.val[0];

        uchar G = src_data.val[1];

        uchar R = src_data.val[2];

        //RGB-->YCbCr颜色空间转换

        uchar y  = (76 * R + 150 * G + 29 * B) >> 8;

        uchar cb = ((128*B -43*R - 85*G)>>8) + 128 ;

        uchar cr = ((128*R -107*G - 21 * B)>>8)+ 128 ;

        //肤色区域判定

        if (y > y_lower && y < y_upper && cb > cb_lower && cb < cb_upper && cr > cr_lower && cr < cr_upper)

         skin_region = 1;

        else

            skin_region = 0;

        uchar temp0= (skin_region == 1)? (uchar)255: B;

        uchar temp1= (skin_region == 1)? (uchar)255: G;

        uchar temp2= (skin_region == 1)? (uchar)255: R;

        dst_data.val[0] = temp0;

        dst_data.val[1] = temp1;

        dst_data.val[2] = temp2;

dst << dst_data;

    }

}

}

void ImgProcess_Top(AXI_STREAM& input, AXI_STREAM& output,int rows, int cols,

            int y_lower,int y_upper,int cb_lower,int cb_upper,int cr_lower,int cr_upper)

{

RGB_IMAGE img_0(rows, cols);

RGB_IMAGE img_1(rows, cols);

#pragma HLS dataflow

hls::AXIvideo2Mat(input,img_0);

hls::hls_skin_dection(img_0,img_1,rows,cols,y_lower,y_upper,cb_lower,cb_upper,cr_lower,cr_upper);

hls::Mat2AXIvideo(img_1, output);

}

#ifndef _TOP_H_

#define _TOP_H_

#include "hls_video.h"

// maximum image size

#define MAX_WIDTH  1920

#define MAX_HEIGHT 1080

typedef unsigned char uchar;

// I/O Image Settings

#define INPUT_IMAGE           "test_1080p.bmp"

#define OUTPUT_IMAGE          "result_1080p.bmp"

#define OUTPUT_IMAGE_GOLDEN   "result_1080p_golden.bmp"

// typedef video library core structures

typedef hls::stream<ap_axiu<24,1,1,1> >               AXI_STREAM;

typedef hls::Mat<MAX_HEIGHT, MAX_WIDTH, HLS_8UC3>     RGB_IMAGE;

typedef hls::Scalar<3, unsigned char>                 RGB_PIXEL;

//声明hls命名空间

namespace hls

{

void hls_skin_dection(RGB_IMAGE& src, RGB_IMAGE& dst,int rows, int cols,

                      int y_lower,int y_upper,int cb_lower,int cb_upper,int cr_lower,int cr_upper);

}

void ImgProcess_Top(AXI_STREAM& input, AXI_STREAM& output,int rows, int cols,

int y_lower,int y_upper,int cb_lower,int cb_upper,int cr_lower,int cr_upper);

#endif

#include "top.h"

#include "hls_opencv.h"

#include "iostream"

#include <time.h>

using namespace std;

using namespace cv;

int main (int argc, char** argv)

{

//IplImage* src = cvLoadImage(INPUT_IMAGE);

IplImage* src = cvLoadImage("test.jpg");

//IplImage* src = cvLoadImage("test_img1.jpg");

IplImage* dst = cvCreateImage(cvGetSize(src), src->depth, src->nChannels);

AXI_STREAM  src_axi, dst_axi;

IplImage2AXIvideo(src, src_axi);

ImgProcess_Top(src_axi, dst_axi, src->height, src->width,0,255,75,125,131,185);

AXIvideo2IplImage(dst_axi, dst);

cvShowImage("src",src);

cvShowImage("dst_hls",dst);

waitKey(0);

return 0;

}