c++で特定方向の行列の最大値のindexを取得する方法

Question

5クラスを持つ画像(224,224,5)=(H, W, depth)の行列があります。

その画像の5つの位置にアクセスして、一次元の行列を作り、その中から最も大きい値のindexを取得するしたいです。
[1, HW+1, (HW2)+1, (HW3)+1, (HW*4)+1]

自分で書いたコード

//PostProc()の中
std::vector<float> segment;
for (int r=0; r<height; r++) { //loop over rows
    for (int c=0; c<width; c++) { //loop over columns
      for (int d=0; d<depth; d++) { //loop over depth (classes)
          segment = result[d*width*height+r*width+c];
      posit = max_element(segment.bigin(), segment.end());
      }
    }
}

ProstProc()のmax_element()で得たpositはクラス数=5なので、0〜４の範囲になると想定してます。

自分でコードを書いたのですが、いまいち自信がないので、教えてください。よろしくお願いします。

---

工程
1.CalSoftmak()でnew演算子のresultに画像を入れる
3.ProtProc()でresultの上の場所にアクセスして行列を作り、maxの値のindexを取得する

main flow

float *softmax   = new float[outWidth*outHeight*outChannel]; 
float *outTensor = new float[outWidth*outHeight*outChannel]; //画像を格納しているtensor

// Calculate softmax on CPU
int outHeight=224;
int outWidth=224;
int outChannel=5;
CPUCalcSoftmax(outTensor, outHeight, outWidth, outChannel, softmax));

PostProc(outHeight, outWidth, outChannel, softmax)

関数

/* Calculate softmax.*/
void CPUCalcSoftmax(const float *data, int height, int width, int depth, float *result)
{
  assert(data && result);

  float *local_res = new float [height*width*depth];
  for (int r=0; r<height; r++) { //loop over rows
    for (int c=0; c<width; c++) { //loop over columns
      double sum = 0.0f;
      for (int d=0; d<depth; d++) { //loop over depth (classes)
	local_res[d] = exp(data[d*width*height+r*width +c]);
	sum=sum+local_res[d];
      }
      for (int d=0; d<depth; d++) { //loop over depth (classes)
	local_res[d] = local_res[d]/sum;
	result[d*width*height+r*width+c] = local_res[d];
      }
    }
  }
  cout << "output file name : in CPUsoftmax  = " << OUTPUT_DIR + output_filenames << endl;
  delete[] local_res;
}

// post process
void PostProc(int height, int width, int depth, float *result){
  assert(result);

  std::vector<float> segment;
  for (int r=0; r<height; r++) { //loop over rows
    for (int c=0; c<width; c++) { //loop over columns
      for (int d=0; d<depth; d++) { //loop over depth (classes)
          segment = result[d*width*height+r*width+c];
      posit = max_element(segment.bigin(), segment.end());
　　　 // positは0~4のはず
      cout << "max value index" << posit << endl;
}

Comments

[yuki23]

質問は何でしょうか？

[trafalbad]

postproc()の中でresultの中身を5つ格納しています。
質問
・格納するのはvector<float>segmentであってますか？
・取得した後のsegmentのベクトルの中から最大値を取得し、その最大値のindexを取得したいです

よろしくお願いします

[fana]

これってコンパイル通るんですか？

・通る場合　→　じゃあ実際に動かしてみれば妥当か否かはわかるよね？　一体何を訊いている？
・通らない場合　→　自身があるとか無いとか言うのは一体何の話？

[trafalbad]

なにわけわかんないこといってんだお前

[fana]

（ひょっとして「コンパイル通る自信がないです」とかいう話だったのか．）

Answer 1

void PostProc(const float result, int height, int width, int depth){
assert(result);
Mat segMat(height, width, CV_8UC3);
for (int r=0; r<height; r++) { //loop over rows
for (int c=0; c<width; c++) { //loop over columns
float pix_val[CLS];
for (int d=0; d<depth; d++) { //loop over depth (classes)
int i = dwidthheight+rwidth+c;
pix_val[d]= result[i];
}
auto max_ind = max_element(pix_val, pix_val + CLS);
int posit = distance(pix_val, max_ind);
segMat.at<Vec3b>(r, c) = Vec3b(colorB[posit], colorG[posit], colorR[posit]);
}
}
cv::imwrite(OUTPUT_DIR + output_filenames, segMat);
}