mirror of
https://github.com/archlinuxarm/PKGBUILDs.git
synced 2024-11-08 22:45:43 +00:00
28 lines
1.3 KiB
Diff
28 lines
1.3 KiB
Diff
|
diff --git a/modules/dnn/src/cuda4dnn/primitives/normalize_bbox.hpp b/modules/dnn/src/cuda4dnn/primitives/normalize_bbox.hpp
|
||
|
|
index f067dddaa701..77b5ee81a08e 100644
|
||
|
|
--- a/modules/dnn/src/cuda4dnn/primitives/normalize_bbox.hpp
|
||
|
|
+++ b/modules/dnn/src/cuda4dnn/primitives/normalize_bbox.hpp
|
||
|
|
@@ -111,7 +111,7 @@ namespace cv { namespace dnn { namespace cuda4dnn {
|
||
|
|
* or there might be several weights
|
||
|
|
* or we don't have to scale
|
||
|
|
*/
|
||
|
|
- if (weight != 1.0)
|
||
|
|
+ if (weight != static_cast<T>(1.0))
|
||
|
|
{
|
||
|
|
kernels::scale1_with_bias1<T>(stream, output, input, weight, 1.0);
|
||
|
|
}
|
||
|
|
diff --git a/modules/dnn/src/cuda4dnn/primitives/region.hpp b/modules/dnn/src/cuda4dnn/primitives/region.hpp
|
||
|
|
index d22d44214e7b..cd44f061ec98 100644
|
||
|
|
--- a/modules/dnn/src/cuda4dnn/primitives/region.hpp
|
||
|
|
+++ b/modules/dnn/src/cuda4dnn/primitives/region.hpp
|
||
|
|
@@ -121,7 +121,7 @@ namespace cv { namespace dnn { namespace cuda4dnn {
|
||
|
|
new_coords
|
||
|
|
);
|
||
|
|
|
||
|
|
- if (nms_iou_threshold > 0) {
|
||
|
|
+ if (nms_iou_threshold > static_cast<T>(0)) {
|
||
|
|
auto output_mat = output_wrapper->getMutableHostMat();
|
||
|
|
CV_Assert(output_mat.type() == CV_32F);
|
||
|
|
for (int i = 0; i < input.get_axis_size(0); i++) {
|
||
|
|
|