| 123456789101112131415161718192021222324252627282930313233343536373839 |
- // Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved.
- //
- // Licensed under the Apache License, Version 2.0 (the "License");
- // you may not use this file except in compliance with the License.
- // You may obtain a copy of the License at
- //
- // http://www.apache.org/licenses/LICENSE-2.0
- //
- // Unless required by applicable law or agreed to in writing, software
- // distributed under the License is distributed on an "AS IS" BASIS,
- // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- // See the License for the specific language governing permissions and
- // limitations under the License.
- #pragma once
- #include <string>
- #include <vector>
- #include <memory>
- #include <utility>
- #include <ctime>
- #include <numeric>
- #include <algorithm>
- namespace PaddleDetection {
- // Object Detection Result
- struct ObjectResult {
- // Rectangle coordinates of detected object: left, right, top, down
- std::vector<int> rect;
- // Class id of detected object
- int class_id;
- // Confidence of detected object
- float confidence;
- };
- void nms(std::vector<ObjectResult> &input_boxes, float nms_threshold);
- } // namespace PaddleDetection
|
