apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-07 17:41:52 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[Bug Fix] Update RKYOLO code format (#1254)

Browse Source 
* 更新rkyolo细节

* 更新rkyolov5 format
This commit is contained in:
Zheng-Bicheng
2023-02-08 19:46:11 +08:00
committed by GitHub
parent 8e2b0c1502
commit e077c005df
3 changed files with 7 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
fastdeploy/vision/detection/contrib/rknpu2/postprocessor.h
View File

@@ -40,12 +40,12 @@ class FASTDEPLOY_DECL RKYOLOPostprocessor {
std::vector<DetectionResult>* results);
/// Set nms_threshold, default 0.45
void SetNMSThreshold(const float& nms_threshold) {
void SetNMSThreshold(float nms_threshold) {
nms_threshold_ = nms_threshold;
}
/// Set conf_threshold, default 0.25
void SetConfThreshold(const float& conf_threshold) {
void SetConfThreshold(float conf_threshold) {
conf_threshold_ = conf_threshold;
}
@@ -56,21 +56,21 @@ class FASTDEPLOY_DECL RKYOLOPostprocessor {
float GetNMSThreshold() const { return nms_threshold_; }
/// Set height and weight
void SetHeightAndWeight(int& height, int& width) {
void SetHeightAndWeight(int height,int width) {
height_ = height;
width_ = width;
}
/// Set pad_hw_values
void SetPadHWValues(std::vector<std::vector<int>> pad_hw_values) {
void SetPadHWValues(const std::vector<std::vector<int>>& pad_hw_values) {
pad_hw_values_ = pad_hw_values;
}
/// Set scale
void SetScale(std::vector<float> scale) { scale_ = scale; }
void SetScale(const std::vector<float>& scale) { scale_ = scale; }
/// Set Anchor
void SetAnchor(std::vector<int> anchors, int anchor_per_branch) {
void SetAnchor(const std::vector<int>& anchors, int anchor_per_branch) {
anchors_ = anchors;
anchor_per_branch_ = anchor_per_branch;
}