Align fastdeploy prediction precision with yolov5 (#11)

* Align fastdeploy prediction precision with yolov5

* Change name of Sort function to SortDetectionResult

* Add stride max_wh is_mini_pad property in __init__.py and unify format of getting image width and length

* Change mergesort.cc to sort_det_res.cc
This commit is contained in:
huangjianhui
2022-07-08 15:52:22 +08:00
committed by GitHub
parent 2b51f0efbc
commit 7d13491879
11 changed files with 151 additions and 65 deletions

View File

@@ -14,11 +14,11 @@
#pragma once
#include <set>
#include <vector>
#include "fastdeploy/core/fd_tensor.h"
#include "fastdeploy/utils/utils.h"
#include "fastdeploy/vision/common/result.h"
#include <set>
#include <vector>
namespace fastdeploy {
namespace vision {
@@ -53,6 +53,9 @@ std::vector<int32_t> TopKIndices(const T* array, int array_size, int topk) {
void NMS(DetectionResult* output, float iou_threshold = 0.5);
} // namespace utils
} // namespace vision
} // namespace fastdeploy
// MergeSort
void SortDetectionResult(DetectionResult* output);
} // namespace utils
} // namespace vision
} // namespace fastdeploy