Skip to content

Commit 564a2bf

Browse files
committed
Fix types
1 parent 74378d0 commit 564a2bf

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

model_api/cpp/models/src/detection_model_yolo.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -620,7 +620,7 @@ std::unique_ptr<ResultBase> YoloV8::postprocess(InferenceResult& infResult) {
620620
bool agnostic = false;
621621
float max_wh = 7680;
622622
std::vector<Anchor> boxes_with_class{boxes}; // TODO: update
623-
for (int i = 0; i < boxes_with_class.size(); ++i) {
623+
for (size_t i = 0; i < boxes_with_class.size(); ++i) {
624624
boxes_with_class[i].left += max_wh * labelIDs[i];
625625
boxes_with_class[i].top += max_wh * labelIDs[i];
626626
boxes_with_class[i].right += max_wh * labelIDs[i];

model_api/cpp/utils/include/utils/nms.hpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -95,5 +95,5 @@ std::vector<size_t> nms(const std::vector<Anchor>& boxes, const std::vector<floa
9595
return keep;
9696
}
9797

98-
std::vector<int> multiclass_nms(const std::vector<AnchorLabeled>& boxes, const std::vector<float>& scores,
98+
std::vector<size_t> multiclass_nms(const std::vector<AnchorLabeled>& boxes, const std::vector<float>& scores,
9999
const float iou_threshold=0.45f, bool includeBoundaries=false, size_t maxNum=200);

model_api/cpp/utils/src/nms.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
#include "utils/nms.hpp"
2020

2121

22-
std::vector<int> multiclass_nms(const std::vector<AnchorLabeled>& boxes, const std::vector<float>& scores,
22+
std::vector<size_t> multiclass_nms(const std::vector<AnchorLabeled>& boxes, const std::vector<float>& scores,
2323
const float iou_threshold, bool includeBoundaries, size_t maxNum) {
2424
std::vector<Anchor> boxes_copy;
2525
boxes_copy.reserve(boxes.size());

0 commit comments

Comments
 (0)