ch997674005
ch997674005
采纳率0%
2019-05-06 16:47 阅读 352

关于在tensorflow中训练模型然后导入sdsoc软件里

5

在tensorflow中训练好了模型,现在想把模型导入sdsoc中,但是权值是一个4维矩阵,我现在想把它压缩成一维矩阵,然后在sdsoc中恢复成原来的矩阵,总是失败,希望有大佬来帮帮我,现在已经完成的是压缩成一维矩阵,但是不知道数据的排列顺序。

for(int batch_cnt=0; batch_cnt<image_Batch; batch_cnt++) {
        ROW_K:
        for(int row_k=0;row_k<5;row_k++){
            COL_K:
            for(int col_k=0;col_k<5;col_k++){
                ROW :
                for (int row = 0; row < CONV_1_OUTPUT_WH; row++) {
                    COL :
                    for (int col = 0; col < CONV_1_OUTPUT_WH; col++) {
                     #pragma HLS PIPELINE II=1
                        float input_pixel = IBRAM[batch_cnt][row+row_k][col+col_k];
                        float mult[8];
                        #pragma HLS array_partition variable=mult complete dim=0
                        D_OUT:
                        for(int co=0;co<8;co++){
                        #pragma HLS unroll
                            mult[co] = input_pixel*WBRAM[row_k][col_k][co];
                            if(row_k==0&&col_k==0)
                                OBRAM[batch_cnt][co][row*CONV_1_OUTPUT_WH+col] = mult[co];
                            else
                                OBRAM[batch_cnt][co][row*CONV_1_OUTPUT_WH+col] += mult[co];
                        }
                    }
                }
            }
        }
    }

  • 点赞
  • 写回答
  • 关注问题
  • 收藏
  • 复制链接分享

1条回答 默认 最新

相关推荐