fix YOLOV5Head naming

open-mmlab · Jan 16, 2023 · 6a43488 · 6a43488
1 parent 41d806d
commit 6a43488
Show file tree

Hide file tree

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