fix potential out-of-border access in gpu StereoBeliefPropagation

This commit is contained in:
Vladislav Vinogradov
2015-08-27 16:09:37 +03:00
parent 94cf5430d0
commit 7ddc0bdd37
2 changed files with 7 additions and 5 deletions

View File

@@ -213,8 +213,8 @@ namespace
if (rthis.levels > 1)
{
int less_rows = (rows + 1) / 2;
int less_cols = (cols + 1) / 2;
int less_rows = rows / 2;
int less_cols = cols / 2;
u2.create(less_rows * rthis.ndisp, less_cols, rthis.msg_type);
d2.create(less_rows * rthis.ndisp, less_cols, rthis.msg_type);
@@ -283,8 +283,8 @@ namespace
for (int i = 1; i < rthis.levels; ++i)
{
cols_all[i] = (cols_all[i-1] + 1) / 2;
rows_all[i] = (rows_all[i-1] + 1) / 2;
cols_all[i] = cols_all[i-1] / 2;
rows_all[i] = rows_all[i-1] / 2;
datas[i].create(rows_all[i] * rthis.ndisp, cols_all[i], rthis.msg_type);