Home
last modified time | relevance | path

Searched refs:num_priors (Results 1 – 17 of 17) sorted by relevance

/dports/graphics/opencv/opencv-4.5.3/modules/dnn/src/cuda4dnn/primitives/
H A Ddetection_output.hpp33 std::size_t num_priors; member
76 num_priors = config.num_priors; in DetectionOutputOp()
94 classwise_topK = num_priors; in DetectionOutputOp()
95 if (keepTopK > 0 && keepTopK < num_priors) in DetectionOutputOp()
103 builder.require<T>(batch_size * num_priors * num_loc_classes * 4); /* decoded boxes */ in DetectionOutputOp()
104 builder.require<T>(batch_size * num_classes * num_priors); /* transposed scores */ in DetectionOutputOp()
139 scores.reshape(-1, num_priors, num_classes); in forward()
153 locations.reshape(batch_size, num_priors, num_loc_classes, 4); in forward()
178 auto shape = std::vector<std::size_t>{batch_size, num_priors, num_loc_classes, 4}; in forward()
192 auto shape = std::vector<std::size_t>{batch_size, num_classes, num_priors}; in forward()
[all …]
H A Dprior_box.hpp34 std::size_t num_priors; member
86 num_priors = config.num_priors; in PriorBoxOp()
114 …variance, num_priors, feature_map_width, feature_map_height, image_width, image_height, normalize,… in forward()
129 std::size_t num_priors; member in cv::dnn::cuda4dnn::PriorBoxOp
/dports/graphics/opencv/opencv-4.5.3/modules/dnn/src/cuda/
H A Ddetection_output.cu40 const size_type num_priors = priors.size() / PRIOR_BOX_SIZE; in decode_bbox() local
47 const auto boxes_per_batch = num_priors * num_loc_classes; in decode_bbox()
204 for (auto i : block_stride_range<BLOCK_SIZE>(num_priors)) in __launch_bounds__()
304 for (auto i : block_stride_range<BLOCK_SIZE>(num_priors)) in __launch_bounds__()
632 auto c = score_id / num_priors; in consolidate_detections()
633 auto prior_id = score_id % num_priors; in consolidate_detections()
641 bbox_id = b * num_priors + prior_id; in consolidate_detections()
730 const auto num_priors = scores.get_axis_size(2); in findTopK() local
766 const auto num_priors = decoded_bboxes.get_axis_size(1); in box_collect() local
850 auto num_priors = scores.get_axis_size(2); in nms_collect() local
[all …]
/dports/graphics/opencv/opencv-4.5.3/modules/dnn/src/opencl/
H A Ddetection_output.cl49 const int num_priors,
61 const int p = ((index / 4 / num_loc_classes) % num_priors) * 4;
74 const int start_index = num_priors * 4 + p;
121 const int num_priors,
133 const int p = ((index / 4 / num_loc_classes) % num_priors) * 4;
146 const int start_index = num_priors * 4 + p;
/dports/graphics/opencv/opencv-4.5.3/modules/dnn/src/vkcom/src/
H A Dop_prior_box.cpp36 int num_priors, in OpPriorBox() argument
46 num_priors_ = num_priors; in OpPriorBox()
/dports/textproc/dbacl/dbacl-1.14.1/src/
H A Drisk-parser.y174 } else if(spec.num_priors < MAX_CAT) { in add_prior_weight()
175 spec.prior[spec.num_priors++] = log(w); in add_prior_weight()
H A Dbayesol.h73 category_count_t num_priors; member
H A Dbayesol.c129 spec.num_priors = 0; in init_spec()
510 } else if( spec.num_cats != spec.num_priors ) { in read_riskspec()
H A Drisk-parser.c1784 } else if(spec.num_priors < MAX_CAT) { in add_prior_weight()
1785 spec.prior[spec.num_priors++] = log(w); in add_prior_weight()
/dports/graphics/opencv/opencv-4.5.3/modules/dnn/src/vkcom/include/
H A Dop_prior_box.hpp24 int num_priors,
/dports/graphics/waifu2x-ncnn-vulkan/waifu2x-ncnn-vulkan-20210521/src/ncnn/examples/
H A Dyolact.cpp156 int num_priors = confidence.h; in detect_yolact() local
159 ncnn::Mat priorbox(4, num_priors); in detect_yolact()
216 for (int i = 0; i < num_priors; i++) in detect_yolact()
/dports/benchmarks/vkpeak/vkpeak-20210430/ncnn/examples/
H A Dyolact.cpp156 int num_priors = confidence.h; in detect_yolact() local
159 ncnn::Mat priorbox(4, num_priors); in detect_yolact()
216 for (int i = 0; i < num_priors; i++) in detect_yolact()
/dports/misc/ncnn/ncnn-20211208/examples/
H A Dyolact.cpp160 int num_priors = confidence.h; in detect_yolact() local
163 ncnn::Mat priorbox(4, num_priors); in detect_yolact()
220 for (int i = 0; i < num_priors; i++) in detect_yolact()
/dports/graphics/vapoursynth-waifu2x-ncnn-vulkan/vapoursynth-waifu2x-ncnn-vulkan-r4/deps/ncnn/examples/
H A Dyolact.cpp156 int num_priors = confidence.h; in detect_yolact() local
159 ncnn::Mat priorbox(4, num_priors); in detect_yolact()
216 for (int i = 0; i < num_priors; i++) in detect_yolact()
/dports/graphics/realsr-ncnn-vulkan/realsr-ncnn-vulkan-20210210/src/ncnn/examples/
H A Dyolact.cpp156 int num_priors = confidence.h; in detect_yolact() local
159 ncnn::Mat priorbox(4, num_priors); in detect_yolact()
216 for (int i = 0; i < num_priors; i++) in detect_yolact()
/dports/graphics/opencv/opencv-4.5.3/modules/dnn/src/layers/
H A Dprior_box_layer.cpp659 config.num_priors = _numPriors; in initCUDA()
H A Ddetection_output_layer.cpp982 config.num_priors = priors_shape[2] / 4; in initCUDA()