-
Notifications
You must be signed in to change notification settings - Fork 5
/
cuSIFT_D.h
45 lines (33 loc) · 1.28 KB
/
cuSIFT_D.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
33
34
35
36
37
38
39
40
41
42
43
44
45
//********************************************************//
// CUDA SIFT extractor by Marten Bjorkman aka Celebrandil //
//********************************************************//
#ifndef CUSIFTD_H
#define CUSIFTD_H
#define NUM_SCALES 5
// Scale down thread block width
#define SCALEDOWN_W 160
// Scale down thread block height
#define SCALEDOWN_H 16
// Find point thread block width
#define MINMAX_W 126
// Find point thread block height
#define MINMAX_H 4
// Laplace thread block width
#define LAPLACE_W 56
// Number of laplace scales
#define LAPLACE_S (NUM_SCALES+3)
// Laplace filter kernel radius
#define LAPLACE_R 4
//====================== Number of threads ====================//
// ScaleDown: SCALEDOWN_W + 4
// LaplaceMulti: (LAPLACE_W+2*LAPLACE_R)*LAPLACE_S
// FindPointsMulti: MINMAX_W + 2
// ComputeOrientations: 128
// ExtractSiftDescriptors: 256
//====================== Number of blocks ====================//
// ScaleDown: (width/SCALEDOWN_W) * (height/SCALEDOWN_H)
// LaplceMulti: (width+2*LAPLACE_R)/LAPLACE_W * height
// FindPointsMulti: (width/MINMAX_W)*NUM_SCALES * (height/MINMAX_H)
// ComputeOrientations: numpts
// ExtractSiftDescriptors: numpts
#endif