aboutsummaryrefslogtreecommitdiff
path: root/tests/validation/fixtures/WarpPerspectiveFixture.h
diff options
context:
space:
mode:
Diffstat (limited to 'tests/validation/fixtures/WarpPerspectiveFixture.h')
-rw-r--r--tests/validation/fixtures/WarpPerspectiveFixture.h14
1 files changed, 8 insertions, 6 deletions
diff --git a/tests/validation/fixtures/WarpPerspectiveFixture.h b/tests/validation/fixtures/WarpPerspectiveFixture.h
index c77efbdc0d..1a6d6c035e 100644
--- a/tests/validation/fixtures/WarpPerspectiveFixture.h
+++ b/tests/validation/fixtures/WarpPerspectiveFixture.h
@@ -70,9 +70,9 @@ public:
protected:
template <typename U>
- void fill(U &&tensor, int i = 0)
+ void fill(U &&tensor)
{
- library->fill_tensor_uniform(tensor, i);
+ library->fill_tensor_uniform(tensor, 0);
}
TensorType compute_target(const TensorShape &shape, const TensorShape &vmask_shape, const float *matrix, InterpolationPolicy policy, BorderMode border_mode, uint8_t constant_border_value,
@@ -112,19 +112,21 @@ protected:
// Create reference
SimpleTensor<T> src{ shape, data_type };
- SimpleTensor<T> valid_mask{ vmask_shape, data_type };
+
+ // Create the valid mask Tensor
+ _valid_mask = SimpleTensor<T>(shape, data_type);
// Fill reference
- fill(src, 0);
- fill(valid_mask, 1);
+ fill(src);
// Compute reference
- return reference::warp_perspective<T>(src, valid_mask, matrix, policy, border_mode, constant_border_value);
+ return reference::warp_perspective<T>(src, _valid_mask, matrix, policy, border_mode, constant_border_value);
}
TensorType _target{};
SimpleTensor<T> _reference{};
BorderMode _border_mode{};
+ SimpleTensor<T> _valid_mask{};
};
} // namespace validation
} // namespace test