Merge pull request openvinotoolkit#776 from eaidova/ea/yolov8

yolov8 notebook
miaviles · Jan 13, 2023 · 108d86c · 108d86c
2 parents f756c18 + 0d08a35
commit 108d86c
Show file tree

Hide file tree

Showing 4 changed files with 1,588 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -146,6 +146,7 @@ More amazing notebooks here!
 | [227-whisper-subtitles-generation](notebooks/227-whisper-subtitles-generation/)<br> | Generate subtitles for video with OpenAI Whisper and OpenVINO | <img src=https://user-images.githubusercontent.com/29454499/204548693-1304ef33-c790-490d-8a8b-d5766acb6254.png  width=225> |
 | [228-clip-zero-shot-image-classification](notebooks/228-clip-zero-shot-image-classification)<br> | Perform Zero-shot Image Classification with CLIP and OpenVINO | <img src=https://user-images.githubusercontent.com/29454499/207795060-437b42f9-e801-4332-a91f-cc26471e5ba2.png  width=500> |
 | [229-distilbert-sequence-classification](notebooks/229-distilbert-sequence-classification/)<br> | Sequence Classification with OpenVINO | <img src = "https://user-images.githubusercontent.com/95271966/206130638-d9847414-357a-4c79-9ca7-76f4ae5a6d7f.png" width=225> |
+| [230-yolov8-optimization](notebooks/230-yolov8-optimization/)<br> | Optimize YOLOv8 using NNCF PTQ API | <img src = "https://user-images.githubusercontent.com/29454499/212105105-f61c8aab-c1ff-40af-a33f-d0ed1fccc72e.png" width=225> |
 </details>
 </p>
 

diff --git a/notebooks/230-yolov8-optimization/230-yolov8-optimization.ipynb b/notebooks/230-yolov8-optimization/230-yolov8-optimization.ipynb
diff --git a/notebooks/230-yolov8-optimization/README.md b/notebooks/230-yolov8-optimization/README.md
@@ -0,0 +1,24 @@
+# Convert and Optimize YOLOv8 with OpenVINO™
+
+![YOLOv8 results](https://user-images.githubusercontent.com/29454499/212105105-f61c8aab-c1ff-40af-a33f-d0ed1fccc72e.png)
+
+This tutorial explains how to convert and optimize the [YOLOv8](https://github.com/ultralytics/) PyTorch\* model with OpenVNO.
+
+
+## Notebook Contents
+
+This tutorial demonstrates step-by-step instructions on how to run and optimize PyTorch\* Yolo V8 with OpenVINO.
+
+The tutorial consists of the following steps:
+- Prepare PyTorch model
+- Download and prepare dataset
+- Validate original model
+- Convert PyTorch model to OpenVINO IR
+- Validate converted model
+- Prepare and run NNCF Postraining optimization pipeline
+- Compare accuracy of the FP32 and quantized models
+- Compare performance of the FP32 and quantized models
+
+## Installation Instructions
+
+If you have not installed all required dependencies, follow the [Installation Guide](../../README.md).
diff --git a/notebooks/README.md b/notebooks/README.md
@@ -146,6 +146,7 @@ More amazing notebooks here!
 | [227-whisper-subtitles-generation](227-whisper-subtitles-generation/)<br> | Generate subtitles for video with OpenAI Whisper and OpenVINO | <img src=https://user-images.githubusercontent.com/29454499/204548693-1304ef33-c790-490d-8a8b-d5766acb6254.png  width=225> |
 | [228-clip-zero-shot-image-classification](228-clip-zero-shot-image-classification)<br> | Perform Zero-shot Image Classification with CLIP and OpenVINO | <img src=https://user-images.githubusercontent.com/29454499/207795060-437b42f9-e801-4332-a91f-cc26471e5ba2.png  width=500> |
 | [229-distilbert-sequence-classification](229-distilbert-sequence-classification/)<br> | Sequence Classification with OpenVINO | <img src = "https://user-images.githubusercontent.com/95271966/206130638-d9847414-357a-4c79-9ca7-76f4ae5a6d7f.png" width=225> |
+| [230-yolov8-optimization](230-yolov8-optimization/)<br> | Optimize YOLOv8 using NNCF PTQ API | <img src = "https://user-images.githubusercontent.com/29454499/212105105-f61c8aab-c1ff-40af-a33f-d0ed1fccc72e.png" width=225> |
 
 
 </details>