Skip to content
GitLab
Explore
Sign in
Primary navigation
Search or go to…
Project
F
ftl
Manage
Activity
Members
Labels
Plan
Issues
Issue boards
Milestones
Wiki
Requirements
Code
Merge requests
Repository
Branches
Commits
Tags
Repository graph
Compare revisions
Snippets
Locked files
Build
Pipelines
Jobs
Pipeline schedules
Test cases
Artifacts
Deploy
Releases
Package Registry
Container Registry
Model registry
Operate
Environments
Terraform modules
Monitor
Incidents
Service Desk
Analyze
Value stream analytics
Contributor analytics
CI/CD analytics
Repository analytics
Code review analytics
Issue analytics
Insights
Model experiments
Help
Help
Support
GitLab documentation
Compare GitLab plans
Community forum
Contribute to GitLab
Provide feedback
Keyboard shortcuts
?
Snippets
Groups
Projects
Show more breadcrumbs
Nicolas Pope
ftl
Commits
ccfc241d
Commit
ccfc241d
authored
6 years ago
by
Nicolas Pope
Browse files
Options
Downloads
Plain Diff
Resolve merge conflict
parents
40e81912
4adce523
No related branches found
No related tags found
No related merge requests found
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
cv-node/CMakeLists.txt
+1
-1
1 addition, 1 deletion
cv-node/CMakeLists.txt
cv-node/src/algorithms/rtcensus.cu
+19
-7
19 additions, 7 deletions
cv-node/src/algorithms/rtcensus.cu
cv-node/src/main.cpp
+5
-5
5 additions, 5 deletions
cv-node/src/main.cpp
with
25 additions
and
13 deletions
cv-node/CMakeLists.txt
+
1
−
1
View file @
ccfc241d
...
@@ -11,7 +11,7 @@ enable_testing()
...
@@ -11,7 +11,7 @@ enable_testing()
set
(
THREADS_PREFER_PTHREAD_FLAG ON
)
set
(
THREADS_PREFER_PTHREAD_FLAG ON
)
set
(
CMAKE_MODULE_PATH
${
CMAKE_MODULE_PATH
}
"
${
CMAKE_SOURCE_DIR
}
/cmake/"
)
set
(
CMAKE_MODULE_PATH
${
CMAKE_MODULE_PATH
}
"
${
CMAKE_SOURCE_DIR
}
/cmake/"
)
find_package
(
glog REQUIRED
)
#
find_package( glog REQUIRED )
if
(
WIN32
)
if
(
WIN32
)
include_directories
(
${
glog_DIR
}
)
include_directories
(
${
glog_DIR
}
)
endif
(
WIN32
)
endif
(
WIN32
)
...
...
This diff is collapsed.
Click to expand it.
cv-node/src/algorithms/rtcensus.cu
+
19
−
7
View file @
ccfc241d
...
@@ -90,6 +90,8 @@ __global__ void census_kernel(PtrStepSzb l, PtrStepSzb r, uint64_t *census) {
...
@@ -90,6 +90,8 @@ __global__ void census_kernel(PtrStepSzb l, PtrStepSzb r, uint64_t *census) {
*/
*/
__global__
void
disp_kernel
(
float
*
disp_l
,
float
*
disp_r
,
size_t
width
,
size_t
height
,
uint64_t
*
census
,
size_t
ds
)
{
__global__
void
disp_kernel
(
float
*
disp_l
,
float
*
disp_r
,
size_t
width
,
size_t
height
,
uint64_t
*
census
,
size_t
ds
)
{
//extern __shared__ uint64_t cache[];
//extern __shared__ uint64_t cache[];
const
int
gamma
=
100
;
size_t
u
=
(
blockIdx
.
x
*
BLOCK_W
)
+
threadIdx
.
x
+
RADIUS2
;
size_t
u
=
(
blockIdx
.
x
*
BLOCK_W
)
+
threadIdx
.
x
+
RADIUS2
;
size_t
v_start
=
(
blockIdx
.
y
*
ROWSperTHREAD
)
+
RADIUS2
;
size_t
v_start
=
(
blockIdx
.
y
*
ROWSperTHREAD
)
+
RADIUS2
;
...
@@ -115,6 +117,8 @@ __global__ void disp_kernel(float *disp_l, float *disp_r, size_t width, size_t h
...
@@ -115,6 +117,8 @@ __global__ void disp_kernel(float *disp_l, float *disp_r, size_t width, size_t h
uint16_t
last_ham2
=
65535
;
uint16_t
last_ham2
=
65535
;
uint16_t
min_disp1
=
65535
;
uint16_t
min_disp1
=
65535
;
uint16_t
min_disp2
=
65535
;
uint16_t
min_disp2
=
65535
;
uint16_t
min_disp1b
=
65535
;
uint16_t
min_disp2b
=
65535
;
uint16_t
min_before1
=
0
;
uint16_t
min_before1
=
0
;
uint16_t
min_before2
=
0
;
uint16_t
min_before2
=
0
;
uint16_t
min_after1
=
0
;
uint16_t
min_after1
=
0
;
...
@@ -152,6 +156,8 @@ __global__ void disp_kernel(float *disp_l, float *disp_r, size_t width, size_t h
...
@@ -152,6 +156,8 @@ __global__ void disp_kernel(float *disp_l, float *disp_r, size_t width, size_t h
min_before1
=
last_ham1
;
min_before1
=
last_ham1
;
min_disp1
=
hamming1
;
min_disp1
=
hamming1
;
dix1
=
d
;
dix1
=
d
;
}
else
if
(
hamming1
<
min_disp1b
)
{
min_disp1b
=
hamming1
;
}
}
if
(
dix1
==
d
)
min_after1
=
hamming1
;
if
(
dix1
==
d
)
min_after1
=
hamming1
;
last_ham1
=
hamming1
;
last_ham1
=
hamming1
;
...
@@ -160,6 +166,8 @@ __global__ void disp_kernel(float *disp_l, float *disp_r, size_t width, size_t h
...
@@ -160,6 +166,8 @@ __global__ void disp_kernel(float *disp_l, float *disp_r, size_t width, size_t h
min_before2
=
last_ham2
;
min_before2
=
last_ham2
;
min_disp2
=
hamming2
;
min_disp2
=
hamming2
;
dix2
=
d
;
dix2
=
d
;
}
else
if
(
hamming2
<
min_disp2b
)
{
min_disp2b
=
hamming2
;
}
}
if
(
dix2
==
d
)
min_after2
=
hamming2
;
if
(
dix2
==
d
)
min_after2
=
hamming2
;
last_ham2
=
hamming2
;
last_ham2
=
hamming2
;
...
@@ -173,9 +181,15 @@ __global__ void disp_kernel(float *disp_l, float *disp_r, size_t width, size_t h
...
@@ -173,9 +181,15 @@ __global__ void disp_kernel(float *disp_l, float *disp_r, size_t width, size_t h
float
d1
=
fit_parabola
(
dix1
,
min_disp1
,
min_before1
,
min_after1
);
float
d1
=
fit_parabola
(
dix1
,
min_disp1
,
min_before1
,
min_after1
);
float
d2
=
fit_parabola
(
dix2
,
min_disp2
,
min_before2
,
min_after2
);
float
d2
=
fit_parabola
(
dix2
,
min_disp2
,
min_before2
,
min_after2
);
// Confidence filter (25)
disp_l
[
v
*
width
+
u
]
=
d2
;
// TODO choice of gamma to depend on disparity variance
disp_r
[
v
*
width
+
u
]
=
d1
;
// Variance with next option, variance with neighbours, variance with past value
disp_l
[
v
*
width
+
u
]
=
((
min_disp2b
-
min_disp2
)
>=
gamma
)
?
d2
:
INFINITY
;
disp_r
[
v
*
width
+
u
]
=
((
min_disp1b
-
min_disp1
)
>=
gamma
)
?
d1
:
INFINITY
;
// TODO If disparity is 0.0f, perhaps
// Use previous value unless it conflicts with present
// Use neighbour values if texture matches
}
}
}
}
...
@@ -198,10 +212,8 @@ __global__ void consistency_kernel(float *d_sub_l, float *d_sub_r, PtrStepSz<flo
...
@@ -198,10 +212,8 @@ __global__ void consistency_kernel(float *d_sub_l, float *d_sub_r, PtrStepSz<flo
auto
b
=
d_sub_r
[
v
*
w
+
u
-
a
];
auto
b
=
d_sub_r
[
v
*
w
+
u
-
a
];
// TODO, if rejected, use semantic info to borrow neighbour
if
(
abs
(
a
-
b
)
<=
1.0
)
disp
(
v
,
u
)
=
abs
((
a
+
b
)
/
2
);
// was 1.0
else
disp
(
v
,
u
)
=
INFINITY
;
if
(
abs
(
a
-
b
)
<=
1.0
)
disp
(
v
,
u
)
=
abs
((
a
+
b
)
/
2
);
else
disp
(
v
,
u
)
=
0.0
f
;
//}
//}
}
}
...
...
This diff is collapsed.
Click to expand it.
cv-node/src/main.cpp
+
5
−
5
View file @
ccfc241d
...
@@ -158,9 +158,9 @@ int main(int argc, char **argv) {
...
@@ -158,9 +158,9 @@ int main(int argc, char **argv) {
l
=
l
(
rect
);
l
=
l
(
rect
);
// HACK to make bad pixels invisible.
// HACK to make bad pixels invisible.
normalize
(
disparity32F
,
depth32F
,
0
,
255
,
NORM_MINMAX
,
CV_8U
);
//
normalize(disparity32F, depth32F, 0, 255, NORM_MINMAX, CV_8U);
r
=
Mat
(
l
.
size
(),
CV_8UC3
,
Vec3i
(
255
,
255
,
255
));
//
r = Mat(l.size(), CV_8UC3, Vec3i(255,255,255));
l
.
copyTo
(
r
,
depth32F
);
//
l.copyTo(r,depth32F);
// TODO Send RGB+D data somewhere
// TODO Send RGB+D data somewhere
...
@@ -177,11 +177,11 @@ int main(int argc, char **argv) {
...
@@ -177,11 +177,11 @@ int main(int argc, char **argv) {
cv
::
Mat
Q_32F
;
// = (Mat_<double>(4,4) << 1, 0, 0, 0, 0, 1, 0, 0, 0, 0, 1, 0, 0, 0, 0, 1); //(4,4,CV_32F);
cv
::
Mat
Q_32F
;
// = (Mat_<double>(4,4) << 1, 0, 0, 0, 0, 1, 0, 0, 0, 0, 1, 0, 0, 0, 0, 1); //(4,4,CV_32F);
calibrate
.
getQ
().
convertTo
(
Q_32F
,
CV_32F
);
calibrate
.
getQ
().
convertTo
(
Q_32F
,
CV_32F
);
cv
::
Mat_
<
cv
::
Vec3f
>
XYZ
(
disparity32F
.
rows
,
disparity32F
.
cols
);
// Output point cloud
cv
::
Mat_
<
cv
::
Vec3f
>
XYZ
(
disparity32F
.
rows
,
disparity32F
.
cols
);
// Output point cloud
reprojectImageTo3D
(
disparity32F
,
XYZ
,
Q_32F
,
fals
e
);
reprojectImageTo3D
(
disparity32F
,
XYZ
,
Q_32F
,
tru
e
);
//cv::imshow("Points",XYZ);
//cv::imshow("Points",XYZ);
cv
::
viz
::
WCloud
cloud_widget
=
cv
::
viz
::
WCloud
(
XYZ
,
r
);
cv
::
viz
::
WCloud
cloud_widget
=
cv
::
viz
::
WCloud
(
XYZ
,
l
);
cloud_widget
.
setRenderingProperty
(
cv
::
viz
::
POINT_SIZE
,
2
);
cloud_widget
.
setRenderingProperty
(
cv
::
viz
::
POINT_SIZE
,
2
);
/* Rotation using rodrigues */
/* Rotation using rodrigues */
...
...
This diff is collapsed.
Click to expand it.
Preview
0%
Loading
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Save comment
Cancel
Please
register
or
sign in
to comment