Home
last modified time | relevance | path

Searched refs:keepTopK (Results 1 – 15 of 15) sorted by relevance

/dports/graphics/opencv/opencv-4.5.3/modules/dnn/src/cuda4dnn/primitives/
H A Ddetection_output.hpp46 int keepTopK; member
89 keepTopK = config.keepTopK; in DetectionOutputOp()
90 CV_Assert(keepTopK > 0); in DetectionOutputOp()
95 if (keepTopK > 0 && keepTopK < num_priors) in DetectionOutputOp()
96 classwise_topK = keepTopK; in DetectionOutputOp()
115 builder.require<int>(batch_size * keepTopK); /* final kept indices */ in DetectionOutputOp()
237 … auto shape = std::vector<std::size_t>{batch_size, static_cast<std::size_t>(keepTopK)}; in forward()
277 int keepTopK; member in cv::dnn::cuda4dnn::DetectionOutputOp
/dports/graphics/opencv/opencv-4.5.3/modules/dnn/src/cuda/
H A Ddetection_output.cu491 …size_type num_classes, size_type num_priors, size_type classwise_topK, size_type keepTopK, index_t… in __launch_bounds__()
603 if (idx < keepTopK) in __launch_bounds__()
605 kept_indices[b * keepTopK + idx] = c * num_priors + prior_id; in __launch_bounds__()
616 …size_type batch_size, size_type num_classes, size_type num_priors, size_type keepTopK, DevicePtr<i… in consolidate_detections() argument
631 auto score_id = kept_indices[b * keepTopK + i]; in consolidate_detections()
843 auto keepTopK = kept_indices.get_axis_size(1); in nms_collect() local
860 …ces, count, scores, threshold, num_classes, num_priors, classwise_topK, keepTopK, background_class… in nms_collect()
882 auto keepTopK = kept_indices.get_axis_size(1); in consolidate_detections() local
887 CV_Assert(batch_size * keepTopK * 7 == output.size()); in consolidate_detections()
890 auto policy = make_policy(kernel, keepTopK, 0, stream); in consolidate_detections()
[all …]
/dports/misc/mnn/MNN-1.2.0/pymnn/pip_package/MNN/tools/mnn_fb/
H A DDetectionOutput.py95 def DetectionOutputAddKeepTopK(builder, keepTopK): builder.PrependInt32Slot(3, keepTopK, 0) argument
/dports/misc/mnn/MNN-1.2.0/source/backend/cpu/
H A DCPUDetectionOutput.hpp18 …CPUDetectionOutput(Backend *backend, int classCount, float nmsThreshold, int keepTopK, float confi…
H A DCPUDetectionOutput.cpp25 …ctionOutput::CPUDetectionOutput(Backend *backend, int classCount, float nmsThreshold, int keepTopK, in CPUDetectionOutput() argument
30 mKeepTopK(keepTopK), in CPUDetectionOutput()
263 return new CPUDetectionOutput(backend, d->classCount(), d->nmsThresholdold(), d->keepTopK(), in onCreate()
/dports/misc/mnn/MNN-1.2.0/source/shape/
H A DShapeDetectionOutput.cpp22 auto maxNumber = op->main_as_DetectionOutput()->keepTopK(); in onComputeSize()
/dports/misc/mnn/MNN-1.2.0/tools/converter/source/caffe/
H A DDetection.cpp36 detectionOutputT->keepTopK = caffeDetect.keep_top_k(); in run()
/dports/misc/mnn/MNN-1.2.0/schema/default/
H A DCaffeOp.fbs267 keepTopK:int;
/dports/misc/mnn/MNN-1.2.0/schema/current/
H A DCaffeOp_generated.h3043 int32_t keepTopK; member
3054 keepTopK(0), in DetectionOutputT()
3078 int32_t keepTopK() const { in keepTopK() function
3130 void add_keepTopK(int32_t keepTopK) { in add_keepTopK()
3131 fbb_.AddElement<int32_t>(10, keepTopK, 0); in add_keepTopK()
3168 int32_t keepTopK = 0,
3182 builder_.add_keepTopK(keepTopK);
4985 { auto _e = keepTopK(); _o->keepTopK = _e; }; in UnPackTo()
5005 auto _keepTopK = _o->keepTopK; in CreateDetectionOutput()
/dports/www/firefox-esr/firefox-91.8.0/browser/components/newtab/lib/PersonalityProvider/
H A DRecipeExecutor.jsm44 keep_top_k: this.keepTopK,
365 keepTopK(item, config) {
/dports/lang/spidermonkey78/firefox-78.9.0/browser/components/newtab/lib/PersonalityProvider/
H A DRecipeExecutor.jsm44 keep_top_k: this.keepTopK,
365 keepTopK(item, config) {
/dports/www/firefox/firefox-99.0/browser/components/newtab/lib/PersonalityProvider/
H A DRecipeExecutor.jsm44 keep_top_k: this.keepTopK,
365 keepTopK(item, config) {
/dports/mail/thunderbird/thunderbird-91.8.0/browser/components/newtab/lib/PersonalityProvider/
H A DRecipeExecutor.jsm44 keep_top_k: this.keepTopK,
365 keepTopK(item, config) {
/dports/graphics/opencv/opencv-4.5.3/modules/dnn/src/layers/
H A Ddetection_output_layer.cpp995 config.keepTopK = _keepTopK; in initCUDA()
/dports/misc/mnn/MNN-1.2.0/express/
H A DNeuralNetWorkOp.cpp1412 param->keepTopK = keep_top_k; in _DetectionOutput()