Function medianBetweenValidPoints

rtpose_wrapper/apps/rtpose.cpp:361–388

}

double medianBetweenValidPoints(const cv::Mat& Input){
  std::vector<ushort> array;
  if (Input.isContinuous()) {
    array.assign(Input.datastart, Input.dataend);
  } else {
    for (int i = 0; i < Input.rows; ++i) {
      array.insert(array.end(), Input.ptr<ushort>(i),
                   Input.ptr<ushort>(i)+Input.cols);
    }
  }
  // jump over the 0s
  std::vector<ushort> array_with_no_zero;
  array_with_no_zero.reserve(array.size());
  for(uint i = 0; i < array.size(); ++i)
  {
    if (not fabs(array[i]) < 0.01f)
    {
      array_with_no_zero.push_back(array[i]);
    }
  }
  if (array_with_no_zero.size() == 0) return 0.0f;
  auto nth_position = array_with_no_zero.begin() +
      array_with_no_zero.size() / 2;
  std::nth_element(array_with_no_zero.begin(),
                   nth_position,
                   array_with_no_zero.end());
  return array_with_no_zero[array_with_no_zero.size() * 0.5];
}
double medianBetweenValidPoints(std::vector<double>& array){
  std::vector<double> array2;
  array2.reserve(array.size());