diff --git a/csrc/mmdeploy/codebase/mmdet/yolo_head.cpp b/csrc/mmdeploy/codebase/mmdet/yolo_head.cpp
index fd746d0bc..1920df191 100644
--- 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
index dd3be05e8..918a359aa 100644
--- 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,