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

3
#include <vector>
4
5

class CuspGPUData;
6
class CuspConfidenceMapSolver
7
8
{
public:
9
    CuspConfidenceMapSolver(int width, int height);
10

11
    virtual ~CuspConfidenceMapSolver();
12

13
14
15
16
    virtual void createSystem(const unsigned char* image, int width, int height,
                              float gradientScaling, float alpha, float beta, float gamma);
    virtual void setInitialSolution(const std::vector<float> &val);
    virtual void solve(int iterations);
17

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

20
21
    // Input image data
    int width, height;
22
23
    
private:
24

25
26
    // Matrices and Vectors
    CuspGPUData* gpuData;
27
};