Skip to content

P
Paddle-Lite

PaddlePaddle / Paddle-Lite

通知 331
Star 4
Fork 1
P
Paddle-Lite
提交 cf01b626 编写于 作者: D dingminghui 提交者： MaxwellDing
浏览文件
操作

remove unused file

上级 3cd728dc
显示空白变更内容
内联 并排
Showing with 0 addition and 451 deletion
out 已删除 100644 → 0
浏览文件 @ 3cd728dc
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 1) // Copyright (c) 2019 PaddlePaddle Authors. All Rights Reserved.
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 2) //
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 3) // Licensed under the Apache License, Version 2.0 (the "License");
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 4) // you may not use this file except in compliance with the License.
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 5) // You may obtain a copy of the License at
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 6) //
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 7) // http://www.apache.org/licenses/LICENSE-2.0
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 8) //
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 9) // Unless required by applicable law or agreed to in writing, software
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 10) // distributed under the License is distributed on an "AS IS" BASIS,
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 11) // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 12) // See the License for the specific language governing permissions and
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 13) // limitations under the License.
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 14)
de43e479 lite/kernels/host/multiclass_nms_compute.cc (Wilber 2019-08-29 21:24:46 +0800 15) #include "lite/kernels/host/multiclass_nms_compute.h"
de43e479 lite/kernels/host/multiclass_nms_compute.cc (Wilber 2019-08-29 21:24:46 +0800 16) #include <map>
de43e479 lite/kernels/host/multiclass_nms_compute.cc (Wilber 2019-08-29 21:24:46 +0800 17) #include <utility>
de43e479 lite/kernels/host/multiclass_nms_compute.cc (Wilber 2019-08-29 21:24:46 +0800 18) #include <vector>
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 19)
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 20) namespace paddle {
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 21) namespace lite {
de43e479 lite/kernels/host/multiclass_nms_compute.cc (Wilber 2019-08-29 21:24:46 +0800 22) namespace kernels {
de43e479 lite/kernels/host/multiclass_nms_compute.cc (Wilber 2019-08-29 21:24:46 +0800 23) namespace host {
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 24)
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 25) template <class T>
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 26) bool SortScorePairDescend(const std::pair<float, T>& pair1,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 27) const std::pair<float, T>& pair2) {
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 28) return pair1.first > pair2.first;
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 29) }
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 30)
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 31) template <class T>
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 32) static void GetMaxScoreIndex(const std::vector<T>& scores,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 33) const T threshold,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 34) int top_k,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 35) std::vector<std::pair<T, int>>* sorted_indices) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 36) for (size_t i = 0; i < scores.size(); ++i) {
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 37) if (scores[i] > threshold) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 38) sorted_indices->push_back(std::make_pair(scores[i], i));
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 39) }
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 40) }
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 41) // Sort the score pair according to the scores in descending order
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 42) std::stable_sort(sorted_indices->begin(),
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 43) sorted_indices->end(),
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 44) SortScorePairDescend<int>);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 45) // Keep top_k scores if needed.
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 46) if (top_k > -1 && top_k < static_cast<int>(sorted_indices->size())) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 47) sorted_indices->resize(top_k);
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 48) }
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 49) }
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 50)
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 51) template <class T>
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 52) static T BBoxArea(const T* box, const bool normalized) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 53) if (box[2] < box[0] || box[3] < box[1]) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 54) // If coordinate values are is invalid
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 55) // (e.g. xmax < xmin or ymax < ymin), return 0.
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 56) return static_cast<T>(0.);
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 57) } else {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 58) const T w = box[2] - box[0];
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 59) const T h = box[3] - box[1];
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 60) if (normalized) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 61) return w * h;
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 62) } else {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 63) // If coordinate values are not within range [0, 1].
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 64) return (w + 1) * (h + 1);
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 65) }
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 66) }
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 67) }
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 68)
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 69) template <class T>
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 70) static T JaccardOverlap(const T* box1, const T* box2, const bool normalized) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 71) if (box2[0] > box1[2] || box2[2] < box1[0] || box2[1] > box1[3] ||
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 72) box2[3] < box1[1]) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 73) return static_cast<T>(0.);
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 74) } else {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 75) const T inter_xmin = std::max(box1[0], box2[0]);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 76) const T inter_ymin = std::max(box1[1], box2[1]);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 77) const T inter_xmax = std::min(box1[2], box2[2]);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 78) const T inter_ymax = std::min(box1[3], box2[3]);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 79) T norm = normalized ? static_cast<T>(0.) : static_cast<T>(1.);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 80) T inter_w = inter_xmax - inter_xmin + norm;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 81) T inter_h = inter_ymax - inter_ymin + norm;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 82) const T inter_area = inter_w * inter_h;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 83) const T bbox1_area = BBoxArea<T>(box1, normalized);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 84) const T bbox2_area = BBoxArea<T>(box2, normalized);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 85) return inter_area / (bbox1_area + bbox2_area - inter_area);
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 86) }
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 87) }
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 88)
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 89) template <class T>
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 90) T PolyIoU(const T* box1,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 91) const T* box2,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 92) const size_t box_size,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 93) const bool normalized) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 94) LOG(FATAL) << "PolyIoU not implement.";
b80194db lite/kernels/host/multiclass_nms_compute.cc (huzhiqiang 2020-04-03 20:18:11 +0800 95) return *box1;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 96) }
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 97)
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 98) template <class T>
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 99) void SliceOneClass(const Tensor& items,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 100) const int class_id,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 101) Tensor* one_class_item) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 102) T* item_data = one_class_item->mutable_data<T>();
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 103) const T* items_data = items.data<T>();
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 104) const int64_t num_item = items.dims()[0];
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 105) const int64_t class_num = items.dims()[1];
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 106) if (items.dims().size() == 3) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 107) int64_t item_size = items.dims()[2];
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 108) for (int i = 0; i < num_item; ++i) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 109) std::memcpy(item_data + i * item_size,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 110) items_data + i * class_num * item_size + class_id * item_size,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 111) sizeof(T) * item_size);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 112) }
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 113) } else {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 114) for (int i = 0; i < num_item; ++i) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 115) item_data[i] = items_data[i * class_num + class_id];
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 116) }
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 117) }
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 118) }
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 119)
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 120) template <typename T>
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 121) void NMSFast(const Tensor& bbox,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 122) const Tensor& scores,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 123) const T score_threshold,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 124) const T nms_threshold,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 125) const T eta,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 126) const int64_t top_k,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 127) std::vector<int>* selected_indices,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 128) const bool normalized) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 129) // The total boxes for each instance.
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 130) int64_t num_boxes = bbox.dims()[0];
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 131) // 4: [xmin ymin xmax ymax]
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 132) // 8: [x1 y1 x2 y2 x3 y3 x4 y4]
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 133) // 16, 24, or 32: [x1 y1 x2 y2 ... xn yn], n = 8, 12 or 16
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 134) int64_t box_size = bbox.dims()[1];
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 135)
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 136) std::vector<T> scores_data(num_boxes);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 137) std::copy_n(scores.data<T>(), num_boxes, scores_data.begin());
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 138) std::vector<std::pair<T, int>> sorted_indices;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 139) GetMaxScoreIndex(scores_data, score_threshold, top_k, &sorted_indices);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 140)
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 141) selected_indices->clear();
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 142) T adaptive_threshold = nms_threshold;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 143) const T* bbox_data = bbox.data<T>();
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 144)
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 145) while (sorted_indices.size() != 0) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 146) const int idx = sorted_indices.front().second;
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 147) bool keep = true;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 148) for (size_t k = 0; k < selected_indices->size(); ++k) {
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 149) if (keep) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 150) const int kept_idx = (*selected_indices)[k];
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 151) T overlap = T(0.);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 152) // 4: [xmin ymin xmax ymax]
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 153) if (box_size == 4) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 154) overlap = JaccardOverlap<T>(bbox_data + idx * box_size,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 155) bbox_data + kept_idx * box_size,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 156) normalized);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 157) }
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 158) // 8: [x1 y1 x2 y2 x3 y3 x4 y4] or 16, 24, 32
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 159) if (box_size == 8 || box_size == 16 || box_size == 24 ||
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 160) box_size == 32) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 161) overlap = PolyIoU<T>(bbox_data + idx * box_size,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 162) bbox_data + kept_idx * box_size,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 163) box_size,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 164) normalized);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 165) }
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 166) keep = overlap <= adaptive_threshold;
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 167) } else {
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 168) break;
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 169) }
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 170) }
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 171) if (keep) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 172) selected_indices->push_back(idx);
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 173) }
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 174) sorted_indices.erase(sorted_indices.begin());
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 175) if (keep && eta < 1 && adaptive_threshold > 0.5) {
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 176) adaptive_threshold *= eta;
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 177) }
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 178) }
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 179) }
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 180)
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 181) template <typename T>
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 182) void MultiClassNMS(const operators::MulticlassNmsParam& param,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 183) const Tensor& scores,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 184) const Tensor& bboxes,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 185) const int scores_size,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 186) std::map<int, std::vector<int>>* indices,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 187) int* num_nmsed_out) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 188) int64_t background_label = param.background_label;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 189) int64_t nms_top_k = param.nms_top_k;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 190) int64_t keep_top_k = param.keep_top_k;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 191) bool normalized = param.normalized;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 192) T nms_threshold = static_cast<T>(param.nms_threshold);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 193) T nms_eta = static_cast<T>(param.nms_eta);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 194) T score_threshold = static_cast<T>(param.score_threshold);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 195)
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 196) int num_det = 0;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 197)
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 198) int64_t class_num = scores_size == 3 ? scores.dims()[0] : scores.dims()[1];
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 199) Tensor bbox_slice, score_slice;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 200) for (int64_t c = 0; c < class_num; ++c) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 201) if (c == background_label) continue;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 202) if (scores_size == 3) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 203) score_slice = scores.Slice<T>(c, c + 1);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 204) bbox_slice = bboxes;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 205) } else {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 206) score_slice.Resize({scores.dims()[0], 1});
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 207) bbox_slice.Resize({scores.dims()[0], 4});
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 208) SliceOneClass<T>(scores, c, &score_slice);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 209) SliceOneClass<T>(bboxes, c, &bbox_slice);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 210) }
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 211) NMSFast(bbox_slice,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 212) score_slice,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 213) score_threshold,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 214) nms_threshold,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 215) nms_eta,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 216) nms_top_k,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 217) &((*indices)[c]),
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 218) normalized);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 219) if (scores_size == 2) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 220) std::stable_sort((*indices)[c].begin(), (*indices)[c].end());
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 221) }
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 222) num_det += (*indices)[c].size();
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 223) }
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 224)
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 225) *num_nmsed_out = num_det;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 226) const T* scores_data = scores.data<T>();
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 227) if (keep_top_k > -1 && num_det > keep_top_k) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 228) const T* sdata;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 229) std::vector<std::pair<float, std::pair<int, int>>> score_index_pairs;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 230) for (const auto& it : *indices) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 231) int label = it.first;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 232) if (scores_size == 3) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 233) sdata = scores_data + label * scores.dims()[1];
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 234) } else {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 235) score_slice.Resize({scores.dims()[0], 1});
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 236) SliceOneClass<T>(scores, label, &score_slice);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 237) sdata = score_slice.data<T>();
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 238) }
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 239) const std::vector<int>& label_indices = it.second;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 240) for (size_t j = 0; j < label_indices.size(); ++j) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 241) int idx = label_indices[j];
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 242) score_index_pairs.push_back(
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 243) std::make_pair(sdata[idx], std::make_pair(label, idx)));
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 244) }
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 245) }
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 246) // Keep top k results per image.
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 247) std::stable_sort(score_index_pairs.begin(),
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 248) score_index_pairs.end(),
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 249) SortScorePairDescend<std::pair<int, int>>);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 250) score_index_pairs.resize(keep_top_k);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 251)
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 252) // Store the new indices.
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 253) std::map<int, std::vector<int>> new_indices;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 254) for (size_t j = 0; j < score_index_pairs.size(); ++j) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 255) int label = score_index_pairs[j].second.first;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 256) int idx = score_index_pairs[j].second.second;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 257) new_indices[label].push_back(idx);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 258) }
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 259) if (scores_size == 2) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 260) for (const auto& it : new_indices) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 261) int label = it.first;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 262) std::stable_sort(new_indices[label].begin(), new_indices[label].end());
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 263) }
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 264) }
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 265) new_indices.swap(*indices);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 266) *num_nmsed_out = keep_top_k;
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 267) }
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 268) }
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 269)
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 270) template <typename T>
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 271) void MultiClassOutput(const Tensor& scores,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 272) const Tensor& bboxes,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 273) const std::map<int, std::vector<int>>& selected_indices,
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 274) const int scores_size,
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 275) Tensor* outs,
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 276) int* oindices = nullptr,
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 277) const int offset = 0) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 278) int64_t class_num = scores.dims()[1];
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 279) int64_t predict_dim = scores.dims()[1];
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 280) int64_t box_size = bboxes.dims()[1];
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 281) if (scores_size == 2) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 282) box_size = bboxes.dims()[2];
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 283) }
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 284) int64_t out_dim = box_size + 2;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 285) auto* scores_data = scores.data<T>();
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 286) auto* bboxes_data = bboxes.data<T>();
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 287) auto* odata = outs->mutable_data<T>();
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 288) const T* sdata;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 289) Tensor bbox;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 290) bbox.Resize({scores.dims()[0], box_size});
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 291) int count = 0;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 292) for (const auto& it : selected_indices) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 293) int label = it.first;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 294) const std::vector<int>& indices = it.second;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 295) if (scores_size == 2) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 296) SliceOneClass<T>(bboxes, label, &bbox);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 297) } else {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 298) sdata = scores_data + label * predict_dim;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 299) }
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 300) for (size_t j = 0; j < indices.size(); ++j) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 301) int idx = indices[j];
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 302) odata[count * out_dim] = label; // label
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 303) const T* bdata;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 304) if (scores_size == 3) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 305) bdata = bboxes_data + idx * box_size;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 306) odata[count * out_dim + 1] = sdata[idx]; // score
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 307) if (oindices != nullptr) {
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 308) oindices[count] = offset + idx;
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 309) }
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 310) } else {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 311) bdata = bbox.data<T>() + idx * box_size;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 312) odata[count * out_dim + 1] = *(scores_data + idx * class_num + label);
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 313) if (oindices != nullptr) {
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 314) oindices[count] = offset + idx * class_num + label;
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 315) }
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 316) }
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 317) // xmin, ymin, xmax, ymax or multi-points coordinates
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 318) std::memcpy(odata + count * out_dim + 2, bdata, box_size * sizeof(T));
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 319) count++;
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 320) }
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 321) }
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 322) }
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 323)
de43e479 lite/kernels/host/multiclass_nms_compute.cc (Wilber 2019-08-29 21:24:46 +0800 324) void MulticlassNmsCompute::Run() {
de43e479 lite/kernels/host/multiclass_nms_compute.cc (Wilber 2019-08-29 21:24:46 +0800 325) auto& param = Param<operators::MulticlassNmsParam>();
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 326) auto* boxes = param.bboxes;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 327) auto* scores = param.scores;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 328) auto* outs = param.out;
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 329) bool return_index = param.index ? true : false;
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 330) auto* index = param.index;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 331) auto score_dims = scores->dims();
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 332) auto score_size = score_dims.size();
de43e479 lite/kernels/host/multiclass_nms_compute.cc (Wilber 2019-08-29 21:24:46 +0800 333)
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 334) std::vector<std::map<int, std::vector<int>>> all_indices;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 335) std::vector<uint64_t> batch_starts = {0};
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 336) int64_t batch_size = score_dims[0];
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 337) int64_t box_dim = boxes->dims()[2];
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 338) int64_t out_dim = box_dim + 2;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 339) int num_nmsed_out = 0;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 340) Tensor boxes_slice, scores_slice;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 341) int n = score_size == 3 ? batch_size : boxes->lod().back().size() - 1;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 342) for (int i = 0; i < n; ++i) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 343) if (score_size == 3) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 344) scores_slice = scores->Slice<float>(i, i + 1);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 345) scores_slice.Resize({score_dims[1], score_dims[2]});
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 346) boxes_slice = boxes->Slice<float>(i, i + 1);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 347) boxes_slice.Resize({score_dims[2], box_dim});
de43e479 lite/kernels/host/multiclass_nms_compute.cc (Wilber 2019-08-29 21:24:46 +0800 348) } else {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 349) auto boxes_lod = boxes->lod().back();
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 350) scores_slice = scores->Slice<float>(boxes_lod[i], boxes_lod[i + 1]);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 351) boxes_slice = boxes->Slice<float>(boxes_lod[i], boxes_lod[i + 1]);
de43e479 lite/kernels/host/multiclass_nms_compute.cc (Wilber 2019-08-29 21:24:46 +0800 352) }
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 353) std::map<int, std::vector<int>> indices;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 354) MultiClassNMS<float>(
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 355) param, scores_slice, boxes_slice, score_size, &indices, &num_nmsed_out);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 356) all_indices.push_back(indices);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 357) batch_starts.push_back(batch_starts.back() + num_nmsed_out);
de43e479 lite/kernels/host/multiclass_nms_compute.cc (Wilber 2019-08-29 21:24:46 +0800 358) }
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 359)
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 360) uint64_t num_kept = batch_starts.back();
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 361) if (num_kept == 0) {
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 362) if (return_index) {
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 363) outs->Resize({0, out_dim});
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 364) index->Resize({0, 1});
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 365) } else {
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 366) outs->Resize({1, 1});
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 367) float* od = outs->mutable_data<float>();
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 368) od[0] = -1;
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 369) batch_starts = {0, 1};
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 370) }
de43e479 lite/kernels/host/multiclass_nms_compute.cc (Wilber 2019-08-29 21:24:46 +0800 371) } else {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 372) outs->Resize({static_cast<int64_t>(num_kept), out_dim});
00000000 lite/kernels/host/multiclass_nms_compute.cc (Not Committed Yet 2020-04-15 02:36:38 +0000 373) <<<<<<< HEAD
99f9b310 lite/kernels/host/multiclass_nms_compute.cc (dingminghui 2020-04-07 16:59:12 +0800 374) (void)outs->mutable_data<float>();
00000000 lite/kernels/host/multiclass_nms_compute.cc (Not Committed Yet 2020-04-15 02:36:38 +0000 375) =======
d571eb4e lite/kernels/host/multiclass_nms_compute.cc (zhupengyang 2020-04-02 10:36:08 +0800 376) outs->mutable_data<float>();
00000000 lite/kernels/host/multiclass_nms_compute.cc (Not Committed Yet 2020-04-15 02:36:38 +0000 377) >>>>>>> upstream/develop
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 378) int offset = 0;
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 379) int* oindices = nullptr;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 380) for (int i = 0; i < n; ++i) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 381) if (score_size == 3) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 382) scores_slice = scores->Slice<float>(i, i + 1);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 383) boxes_slice = boxes->Slice<float>(i, i + 1);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 384) scores_slice.Resize({score_dims[1], score_dims[2]});
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 385) boxes_slice.Resize({score_dims[2], box_dim});
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 386) if (return_index) {
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 387) offset = i * score_dims[2];
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 388) }
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 389) } else {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 390) auto boxes_lod = boxes->lod().back();
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 391) scores_slice = scores->Slice<float>(boxes_lod[i], boxes_lod[i + 1]);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 392) boxes_slice = boxes->Slice<float>(boxes_lod[i], boxes_lod[i + 1]);
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 393) if (return_index) {
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 394) offset = boxes_lod[i] * score_dims[1];
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 395) }
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 396) }
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 397) int64_t s = static_cast<int64_t>(batch_starts[i]);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 398) int64_t e = static_cast<int64_t>(batch_starts[i + 1]);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 399) if (e > s) {
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 400) Tensor out = outs->Slice<float>(s, e);
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 401) if (return_index) {
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 402) index->Resize({static_cast<int64_t>(num_kept), 1});
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 403) int* output_idx = index->mutable_data<int>();
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 404) oindices = output_idx + s;
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 405) }
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 406) MultiClassOutput<float>(scores_slice,
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 407) boxes_slice,
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 408) all_indices[i],
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 409) score_dims.size(),
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 410) &out,
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 411) oindices,
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 412) offset);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 413) }
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 414) }
de43e479 lite/kernels/host/multiclass_nms_compute.cc (Wilber 2019-08-29 21:24:46 +0800 415) }
de43e479 lite/kernels/host/multiclass_nms_compute.cc (Wilber 2019-08-29 21:24:46 +0800 416)
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 417) LoD lod;
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 418) lod.emplace_back(batch_starts);
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 419) if (return_index) {
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 420) index->set_lod(lod);
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 421) }
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 422) outs->set_lod(lod);
deaddf9d lite/kernels/host/multiclass_nms_compute.cc (juncaipeng 2019-09-03 15:39:16 +0800 423) }
de43e479 lite/kernels/host/multiclass_nms_compute.cc (Wilber 2019-08-29 21:24:46 +0800 424) } // namespace host
de43e479 lite/kernels/host/multiclass_nms_compute.cc (Wilber 2019-08-29 21:24:46 +0800 425) } // namespace kernels
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 426) } // namespace lite
699d6cd0 lite/arm/math/multiclass_nms.cc (Yan Chunwei 2019-08-16 22:39:39 +0800 427) } // namespace paddle
de43e479 lite/kernels/host/multiclass_nms_compute.cc (Wilber 2019-08-29 21:24:46 +0800 428)
de43e479 lite/kernels/host/multiclass_nms_compute.cc (Wilber 2019-08-29 21:24:46 +0800 429) REGISTER_LITE_KERNEL(multiclass_nms,
de43e479 lite/kernels/host/multiclass_nms_compute.cc (Wilber 2019-08-29 21:24:46 +0800 430) kHost,
de43e479 lite/kernels/host/multiclass_nms_compute.cc (Wilber 2019-08-29 21:24:46 +0800 431) kFloat,
de43e479 lite/kernels/host/multiclass_nms_compute.cc (Wilber 2019-08-29 21:24:46 +0800 432) kNCHW,
de43e479 lite/kernels/host/multiclass_nms_compute.cc (Wilber 2019-08-29 21:24:46 +0800 433) paddle::lite::kernels::host::MulticlassNmsCompute,
de43e479 lite/kernels/host/multiclass_nms_compute.cc (Wilber 2019-08-29 21:24:46 +0800 434) def)
de43e479 lite/kernels/host/multiclass_nms_compute.cc (Wilber 2019-08-29 21:24:46 +0800 435) .BindInput("BBoxes", {LiteType::GetTensorTy(TARGET(kHost))})
de43e479 lite/kernels/host/multiclass_nms_compute.cc (Wilber 2019-08-29 21:24:46 +0800 436) .BindInput("Scores", {LiteType::GetTensorTy(TARGET(kHost))})
de43e479 lite/kernels/host/multiclass_nms_compute.cc (Wilber 2019-08-29 21:24:46 +0800 437) .BindOutput("Out", {LiteType::GetTensorTy(TARGET(kHost))})
0679feed lite/kernels/host/multiclass_nms_compute.cc (yiicy 2020-02-20 11:39:43 +0800 438) .Finalize();
0679feed lite/kernels/host/multiclass_nms_compute.cc (yiicy 2020-02-20 11:39:43 +0800 439)
0679feed lite/kernels/host/multiclass_nms_compute.cc (yiicy 2020-02-20 11:39:43 +0800 440) REGISTER_LITE_KERNEL(multiclass_nms2,
0679feed lite/kernels/host/multiclass_nms_compute.cc (yiicy 2020-02-20 11:39:43 +0800 441) kHost,
0679feed lite/kernels/host/multiclass_nms_compute.cc (yiicy 2020-02-20 11:39:43 +0800 442) kFloat,
0679feed lite/kernels/host/multiclass_nms_compute.cc (yiicy 2020-02-20 11:39:43 +0800 443) kNCHW,
0679feed lite/kernels/host/multiclass_nms_compute.cc (yiicy 2020-02-20 11:39:43 +0800 444) paddle::lite::kernels::host::MulticlassNmsCompute,
0679feed lite/kernels/host/multiclass_nms_compute.cc (yiicy 2020-02-20 11:39:43 +0800 445) def)
0679feed lite/kernels/host/multiclass_nms_compute.cc (yiicy 2020-02-20 11:39:43 +0800 446) .BindInput("BBoxes", {LiteType::GetTensorTy(TARGET(kHost))})
0679feed lite/kernels/host/multiclass_nms_compute.cc (yiicy 2020-02-20 11:39:43 +0800 447) .BindInput("Scores", {LiteType::GetTensorTy(TARGET(kHost))})
0679feed lite/kernels/host/multiclass_nms_compute.cc (yiicy 2020-02-20 11:39:43 +0800 448) .BindOutput("Out", {LiteType::GetTensorTy(TARGET(kHost))})
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 449) .BindOutput("Index",
e1c4adfd lite/kernels/host/multiclass_nms_compute.cc (yiicy 2019-12-24 16:10:19 +0800 450) {LiteType::GetTensorTy(TARGET(kHost), PRECISION(kInt32))})
de43e479 lite/kernels/host/multiclass_nms_compute.cc (Wilber 2019-08-29 21:24:46 +0800 451) .Finalize();
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或