Searched refs:decoded_bboxes (Results 1 – 2 of 2) sorted by relevance
| /dports/graphics/opencv/opencv-4.5.3/modules/dnn/src/cuda/ |
| H A D | detection_output.cu | 28 __global__ void decode_bbox(Span<T> decoded_bboxes, View<T> locations, View<T> priors, in decode_bbox() argument
45 auto decoded_bboxes_vPtr = vector_type::get_pointer(decoded_bboxes.data()); in decode_bbox()
48 for (auto idx : grid_stride_range(decoded_bboxes.size() / 4)) in decode_bbox()
369 auto decoded_bboxes = decoded_bboxes_.data() + b * num_priors * num_loc_classes * 4; in box_collect() local
378 auto decoded_bboxes_vPtr = vector_type::get_pointer(decoded_bboxes); in box_collect()
619 auto decoded_bboxes_vPtr = vector_type::get_pointer(decoded_bboxes.data()); in consolidate_detections()
672 auto policy = make_policy(kernel, decoded_bboxes.size() / 4, 0, stream); in launch_decode_boxes_kernel()
755 CV_Assert(decoded_bboxes.get_axis_size(0) == batch_size); in box_collect()
766 const auto num_priors = decoded_bboxes.get_axis_size(1); in box_collect()
768 CV_Assert(!share_location || decoded_bboxes.get_axis_size(2) == 1); in box_collect()
[all …]
|
| /dports/graphics/opencv/opencv-4.5.3/modules/dnn/src/cuda4dnn/kernels/ |
| H A D | detection_output.hpp | 25 …eam& stream, csl::TensorSpan<T> collected_bboxes, csl::TensorView<T> decoded_bboxes, csl::TensorVi…
38 …csl::TensorView<T> decoded_bboxes, csl::TensorView<T> scores, bool share_location, csl::DevicePtr<…
|