[fix] Fix component not found issue by changing class YOLOV5Head to…

… `class YOLOv5Head` (#1785)
open-mmlab · Feb 23, 2023 · 2a98db9 · 2a98db9
1 parent 7b45556
commit 2a98db9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/csrc/mmdeploy/codebase/mmdet/yolo_head.cpp b/csrc/mmdeploy/codebase/mmdet/yolo_head.cpp
@@ -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
diff --git a/csrc/mmdeploy/codebase/mmdet/yolo_head.h b/csrc/mmdeploy/codebase/mmdet/yolo_head.h
@@ -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,