DisparityMapEvaluation.h

Go to the documentation of this file.

/*
Copyright (C) 2024 Scott Grauer-Gray
 
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.
 
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.
 
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
*/
 
#ifndef DISPARITY_MAP_EVALUATION_H_
#define DISPARITY_MAP_EVALUATION_H_
 
#include <map>
#include <iostream>
#include <numeric>
#include <array>
#include <vector>
#include "RunEval/RunData.h"
 
namespace beliefprop {
 
constexpr std::string_view kAvgRMSErrorHeader{"Average RMS error"};
 
constexpr std::array<float, 4> kDisparityDiffThresholds{
  0.001, 2.01, 5.01, 10.01};
 
constexpr float kMaxDiffCap{
  kDisparityDiffThresholds[std::size(kDisparityDiffThresholds) - 1]};
 
struct DisparityMapEvaluationParams {
  const std::vector<float> output_diff_thresholds{
    beliefprop::kDisparityDiffThresholds.cbegin(),
    beliefprop::kDisparityDiffThresholds.cend()};
  const float max_diff_cap{beliefprop::kMaxDiffCap};
};
 
};
 
class DisparityMapEvaluation {
public:
  void InitializeWithEvalParams(
    const beliefprop::DisparityMapEvaluationParams& eval_params);
 
  RunData AsRunData() const;
 
  std::array<float, 2> average_disp_abs_diff_no_max_w_max_{0, 0};
  float disparity_error_max_{std::numeric_limits<float>::max()};
 
  std::map<float, float> prop_sig_diff_pixels_at_thresholds_;
 
  std::map<float, unsigned int> num_sig_diff_pixels_at_thresholds_;
};
 
#endif /* DISPARITY_MAP_EVALUATION_H_ */