Method _argmax

sloam/src/segmentation/inference.cpp:318–335

}

void Segmentation::_argmax(const float *in, cv::Mat& maskImg){
  std::vector<unsigned char> max;
  size_t numClasses_ = 2;
  size_t outSize = _img_h * _img_w * numClasses_;
  for (unsigned int i = 0; i < outSize; i += numClasses_) {
    unsigned int maxIdx = i;
    unsigned char outIdx = 0;
    for (unsigned int c = 1; c < numClasses_; c++) {
      if (in[maxIdx] < in[i + c]) {
        maxIdx = i + c;
        outIdx = c;
      }
    }
    if(outIdx == 1) outIdx = 255;
    max.push_back(outIdx);
  }
  memcpy(maskImg.data, max.data(), max.size()*sizeof(unsigned char));
}

void Segmentation::runERF(cv::Mat& rImg, cv::Mat& maskImg){
    cv::Mat pImg(rImg.rows, rImg.cols, CV_32FC2, cv::Scalar(0));