mirrors
/
mmdeploy
mirror of https://github.com/open-mmlab/mmdeploy.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[fix] Fix component not found issue by changing `class YOLOV5Head` to `class YOLOv5Head` (#1785)

pull/1804/head
Yee 2023-02-23 10:10:19 +08:00 committed by GitHub
parent 7b455563dd
commit 2a98db9670
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
csrc/mmdeploy/codebase/mmdet/yolo_head.cpp
View File

@ -208,7 +208,7 @@ std::array<float, 4> YOLOV3Head::yolo_decode(float box_x, float box_y, float box
return std::array<float, 4>{box_x, box_y, box_w, box_h};
}
std::array<float, 4> YOLOV5Head::yolo_decode(float box_x, float box_y, float box_w, float box_h,
std::array<float, 4> YOLOv5Head::yolo_decode(float box_x, float box_y, float box_w, float box_h,
float stride,
const std::vector<std::vector<float>>& anchor, int j,
int i, int a) const {
@ -224,6 +224,6 @@ std::array<float, 4> YOLOV5Head::yolo_decode(float box_x, float box_y, float box
}
MMDEPLOY_REGISTER_CODEBASE_COMPONENT(MMDetection, YOLOV3Head);
MMDEPLOY_REGISTER_CODEBASE_COMPONENT(MMDetection, YOLOV5Head);
MMDEPLOY_REGISTER_CODEBASE_COMPONENT(MMDetection, YOLOv5Head);
} // namespace mmdeploy::mmdet

2
csrc/mmdeploy/codebase/mmdet/yolo_head.h
View File

@ -38,7 +38,7 @@ class YOLOV3Head : public YOLOHead {
int a) const override;
};
class YOLOV5Head : public YOLOHead {
class YOLOv5Head : public YOLOHead {
public:
using YOLOHead::YOLOHead;
std::array<float, 4> yolo_decode(float box_x, float box_y, float box_w, float box_h, float stride,