[Feature] Support yolov8-world and yolov8-oiv7 model

CVHub520 · May 29, 2024 · 98a95db · 98a95db
1 parent 2cd84b6
commit 98a95db
Show file tree

Hide file tree

Showing 18 changed files with 4,066 additions and 15 deletions.
diff --git a/README.md b/README.md
@@ -46,15 +46,15 @@
 ## 🥳 What's New [⏏️](#📄-table-of-contents)
 
 - May. 2024:
-  - Support [YOLOv10](https://github.com/THU-MIG/yolov10) model.
+  - ✨✨✨ Support [YOLOv8-World](https://docs.ultralytics.com/models/yolo-world), [YOLOv8-oiv7](https://docs.ultralytics.com/models/yolov8), [YOLOv10](https://github.com/THU-MIG/yolov10) model.
   - 🤗 Release the latest version [2.3.6](https://github.com/CVHub520/X-AnyLabeling/releases/tag/v2.3.6) 🤗
+  - Add feature to display confidence score.
 - Mar. 2024:
   - Release version [2.3.5](https://github.com/CVHub520/X-AnyLabeling/releases/tag/v2.3.5).
 - Feb. 2024:
   - Release version [2.3.4](https://github.com/CVHub520/X-AnyLabeling/releases/tag/v2.3.4).
   - Enable label display feature.
   - Release version [2.3.3](https://github.com/CVHub520/X-AnyLabeling/releases/tag/v2.3.3).
-  - ✨✨✨ Support [YOLO-World](https://github.com/AILab-CVC/YOLO-World) model.
   - Release version [2.3.2](https://github.com/CVHub520/X-AnyLabeling/releases/tag/v2.3.2).
   - Support [YOLOv9](https://github.com/WongKinYiu/yolov9) model.
   - Support the conversion from a horizontal bounding box to a rotated bounding box.

diff --git a/README_zh-CN.md b/README_zh-CN.md
@@ -45,8 +45,9 @@
 ## 🥳 新功能 [⏏️](#📄-目录)
 
 - 2024年5月：
-  - 支持[YOLOv10](https://github.com/THU-MIG/yolov10)模型.
+  - ✨✨✨ 支持[YOLOv8-World](https://docs.ultralytics.com/models/yolo-world), [YOLOv8-oiv7](https://docs.ultralytics.com/models/yolov8), [YOLOv10](https://github.com/THU-MIG/yolov10)模型。
   - 🤗 发布[2.3.6](https://github.com/CVHub520/X-AnyLabeling/releases/tag/v2.3.6)最新版本 🤗
+  - 支持显示模型预测得分。
 - 2024年3月：
   - 发布[2.3.5](https://github.com/CVHub520/X-AnyLabeling/releases/tag/v2.3.5)版本。
 - 2024年2月：

diff --git a/anylabeling/configs/auto_labeling/models.yaml b/anylabeling/configs/auto_labeling/models.yaml
@@ -138,18 +138,30 @@
   config_file: ":/yolov6x_mbla.yaml"
 - model_name: "yolov7-r20230520"
   config_file: ":/yolov7.yaml"
+- model_name: "yolov8l-oiv7-r20240529"
+  config_file: ":/yolov8l_oiv7.yaml"
 - model_name: "yolov8l-seg-r20230620"
   config_file: ":/yolov8l_seg.yaml"
+- model_name: "yolov8l-worldv2-cc3m-r20240529"
+  config_file: ":/yolov8l_worldv2_cc3m.yaml"
+- model_name: "yolov8l-worldv2-r20240529"
+  config_file: ":/yolov8l_worldv2.yaml"
 - model_name: "yolov8l-r20230520"
   config_file: ":/yolov8l.yaml"
 - model_name: "yolov8m_ocsort-r20231020"
   config_file: ":/yolov8m_ocsort.yaml"
+- model_name: "yolov8m-oiv7-r20240529"
+  config_file: ":/yolov8m_oiv7.yaml"
 - model_name: "yolov8m-seg-r20230620"
   config_file: ":/yolov8m_seg.yaml"
+- model_name: "yolov8m-worldv2-r20240529"
+  config_file: ":/yolov8m_worldv2.yaml"
 - model_name: "yolov8m-r20230520"
   config_file: ":/yolov8m.yaml"
 - model_name: "yolov8n_efficientvit_sam_l0_vit_h-r20231020"
   config_file: ":/yolov8n_efficientvit_sam_l0_vit_h.yaml"
+- model_name: "yolov8n-oiv7-r20240529"
+  config_file: ":/yolov8n_oiv7.yaml"
 - model_name: "yolov8n-pose-r20231103"
   config_file: ":/yolov8n_pose.yaml"
 - model_name: "yolov8n-seg-r20230620"
@@ -160,18 +172,28 @@
   config_file: ":/yolov8s_cls.yaml"
 - model_name: "yolov8s-obb-r20240111"
   config_file: ":/yolov8s_obb.yaml"
+- model_name: "yolov8s-oiv7-r20240529"
+  config_file: ":/yolov8s_oiv7.yaml"
 - model_name: "yolov8s_sahi-r20231008"
   config_file: ":/yolov8s_sahi.yaml"
 - model_name: "yolov8s-seg-r20230620"
   config_file: ":/yolov8s_seg.yaml"
+- model_name: "yolov8s-worldv2-r20240529"
+  config_file: ":/yolov8s_worldv2.yaml"
 - model_name: "yolov8s-r20230520"
   config_file: ":/yolov8s.yaml"
+- model_name: "yolov8x-oiv7-r20240529"
+  config_file: ":/yolov8x_oiv7.yaml"
 - model_name: "yolov8x-pose-p6-r20231103"
   config_file: ":/yolov8x_pose_p6.yaml"
 - model_name: "yolov8x-seg-r20230620"
   config_file: ":/yolov8x_seg.yaml"
+- model_name: "yolov8x-worldv2-r20240529"
+  config_file: ":/yolov8x_worldv2.yaml"
 - model_name: "yolov8x-r20230520"
   config_file: ":/yolov8x.yaml"
+- model_name: "yolov8x6-oiv7-r20240529"
+  config_file: ":/yolov8x6_oiv7.yaml"
 - model_name: "yolov9-gelan-c-r20240224"
   config_file: ":/yolov9_gelan_c.yaml"
 - model_name: "yolov9-gelan-e-r20240224"
@@ -192,8 +214,6 @@
   config_file: ":/yolov10s.yaml"
 - model_name: "yolov10x-r20240525"
   config_file: ":/yolov10x.yaml"
-- model_name: "yolow_l-r20240227"
-  config_file: ":/yolow_l.yaml"
 - model_name: "yolox_l_dwpose_ucoco-r20230820"
   config_file: ":/yolox_l_dwpose_ucoco.yaml"
 - model_name: "yolox_s-r20230520"