-
Notifications
You must be signed in to change notification settings - Fork 0
/
Projection_GPU.h
32 lines (27 loc) · 930 Bytes
/
Projection_GPU.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
#ifndef PROJECTION_GPU_H
#define PROJECTION_GPU_H
#include <cuda.h>
#include <cuda_runtime.h>
#include <opencv2\opencv.hpp>
class DimensionConvertor;
class Projection_GPU{
public:
Projection_GPU(int width, int height, const cv::Mat intrinsic);
~Projection_GPU();
void PlaneProjection(const float4* nd_device, const int* labels_device, const float* variance_device, const float3* points3d_device);
void PlaneProjection(const float3* normals_device, const float3* centers_device, const int* labels_device,
const float* variance_device, const float3* points3d_device);
float3* GetPlaneFitted3D_Host()const;
float3* GetPlaneFitted3D_Device()const;
private:
void initMemory();
void getProjectedMap();
void initNormalized3D();
DimensionConvertor* dim;
int width;
int height;
float3* PlaneFitted3D_Host;
float3* PlaneFitted3D_Device;
float3* Normalized3D_Device;
};
#endif