YOLOv8 OBB model (#1821)

Ticket: CVS-133798
openvinotoolkit · Mar 19, 2024 · 8b75c93 · 8b75c93
1 parent daa7e4b
commit 8b75c93
Show file tree

Hide file tree

Showing 5 changed files with 852 additions and 0 deletions.
diff --git a/.ci/ignore_convert_execution.txt b/.ci/ignore_convert_execution.txt
@@ -14,6 +14,7 @@ notebooks/228-clip-zero-shot-image-classification/228-clip-zero-shot-classificat
 notebooks/230-yolov8-optimization/230-yolov8-instance-segmentation.ipynb
 notebooks/230-yolov8-optimization/230-yolov8-keypoint-detection.ipynb
 notebooks/230-yolov8-optimization/230-yolov8-object-detection.ipynb
+notebooks/230-yolov8-optimization/230-yolov8-obb.ipynb
 notebooks/231-instruct-pix2pix-image-editing/231-instruct-pix2pix-image-editing.ipynb
 notebooks/232-clip-language-saliency-map/232-clip-language-saliency-map.ipynb
 notebooks/233-blip-visual-language-processing/233-blip-visual-language-processing.ipynb

diff --git a/.ci/ignore_pip_conflicts.txt b/.ci/ignore_pip_conflicts.txt
@@ -7,6 +7,7 @@ notebooks/212-pyannote-speaker-diarization/212-pyannote-speaker-diarization.ipyn
 notebooks/230-yolov8-optimization/230-yolov8-instance-segmentation.ipynb  # ultralytics==8.0.43
 notebooks/230-yolov8-optimization/230-yolov8-keypoint-detection.ipynb  # ultralytics==8.0.159
 notebooks/230-yolov8-optimization/230-yolov8-object-detection.ipynb  # ultralytics==8.0.43
+notebooks/230-yolov8-optimization/230-yolov8-obb.ipynb  # ultralytics==8.1.24
 notebooks/254-llm-chatbot/254-llm-chatbot.ipynb # nncf@https://github.com/openvinotoolkit/nncf/tree/release_v280
 notebooks/254-llm-chatbot/254-rag-chatbot.ipynb # nncf@https://github.com/openvinotoolkit/nncf/tree/release_v280
 notebooks/256-bark-text-to-audio/256-bark-text-to-audio.ipynb  # torch==1.13

diff --git a/.ci/spellcheck/.pyspelling.wordlist.txt b/.ci/spellcheck/.pyspelling.wordlist.txt
@@ -184,6 +184,8 @@ DocumentLoaders
 doi
 Dollár
 donut
+DOTA
+DOTAv
 Downloader
 downloader
 downsample
@@ -454,6 +456,8 @@ NSFW
 numpy
 NumPy
 OASST
+OBB
+obb
 ocr
 OCRv
 odometry

diff --git a/notebooks/230-yolov8-optimization/230-yolov8-obb.ipynb b/notebooks/230-yolov8-optimization/230-yolov8-obb.ipynb
diff --git a/notebooks/230-yolov8-optimization/README.md b/notebooks/230-yolov8-optimization/README.md
@@ -14,6 +14,7 @@ This tutorial explains how to convert and optimize the [YOLOv8](https://github.c
 - [230-yolov8-object-detection](./230-yolov8-object-detection.ipynb) [![Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/openvinotoolkit/openvino_notebooks/blob/main/notebooks/230-yolov8-optimization/230-yolov8-object-detection.ipynb)
 - [230-yolov8-instance-segmentation](./230-yolov8-instance-segmentation.ipynb) [![Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/openvinotoolkit/openvino_notebooks/blob/main/notebooks/230-yolov8-optimization/230-yolov8-instance-segmentation.ipynb)
 - [230-yolov8-keypoint-detection](./230-yolov8-keypoint-detection.ipynb) [![Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/openvinotoolkit/openvino_notebooks/blob/main/notebooks/230-yolov8-optimization/230-yolov8-keypoint-detection.ipynb)
+- [230-yolov8-obb](./230-yolov8-obb.ipynb) [![Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/openvinotoolkit/openvino_notebooks/blob/main/notebooks/230-yolov8-optimization/230-yolov8-obb.ipynb)
 
 
 Each case tutorial consists of the following steps::
-Original file line number
+Diff line change
@@ Expand Up / @@ -184,6 +184,8 @@ DocumentLoaders @@
     doi
     Dollár
     donut
+    DOTA
+    DOTAv
     Downloader
     downloader
     downsample
@@ Expand Down Expand Up / @@ -454,6 +456,8 @@ NSFW @@
     numpy
     NumPy
     OASST
+    OBB
+    obb
     ocr
     OCRv
     odometry
@@ Expand Down @@