cm_cusp.h 676 Bytes
Newer Older
1
2
#pragma once

3
#include <vector>
4
5

class CuspGPUData;
6
class CuspConfidenceMapSolver
7
8
9
10
11
12
13
{
public:
	CuspConfidenceMapSolver(int width, int height, float gradientScaling,
	                        float alpha, float beta, float gamma);

	virtual ~CuspConfidenceMapSolver();

14
	virtual void createSystem(const unsigned char* image, int width, int height);
15
16
17
18
19
20
21
22
23
24
25
	virtual void setInitialSolution(const std::vector<float> &val);
	virtual void solve();

	virtual const float* getSolution(int &width, int &height) const;

private:
	// Input image data
	int width, height;

	// Solver parameters
	float gradientScaling;
26
	float alpha, beta, gamma;
27
28
29
30

	// Matrices and Vectors
	CuspGPUData* gpuData;
};