From 8cdfa1e0642a15f0d5f260a152e2571afa1bf3b3 Mon Sep 17 00:00:00 2001 From: Igor Davidyuk Date: Thu, 4 Jul 2024 11:00:02 +0300 Subject: [PATCH] Apply suggestions from code review Changing the visualizer import path Co-authored-by: Ludo Cornelissen --- geti_sdk/deployment/resources/OVMS_README.md | 2 +- notebooks/008_deploy_project.ipynb | 2 +- notebooks/010_model_serving.ipynb | 2 +- notebooks/012_post_inference_hooks.ipynb | 2 +- .../use_cases/101_simulate_low_light_product_inspection.ipynb | 2 +- 5 files changed, 5 insertions(+), 5 deletions(-) diff --git a/geti_sdk/deployment/resources/OVMS_README.md b/geti_sdk/deployment/resources/OVMS_README.md index fb3eee9b..a85fac65 100644 --- a/geti_sdk/deployment/resources/OVMS_README.md +++ b/geti_sdk/deployment/resources/OVMS_README.md @@ -62,7 +62,7 @@ image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB) predictions = deployment.infer(image=image) # Show inference result -from geti_sdk.prediction_visualization.visualizer import Visualizer +from geti_sdk import Visualizer visualizer = Visualizer() result_image = visualizer.draw(image=image, annotation_scene=predictions) visualizer.show_window(result_image) diff --git a/notebooks/008_deploy_project.ipynb b/notebooks/008_deploy_project.ipynb index f4968ac3..30bfd81d 100644 --- a/notebooks/008_deploy_project.ipynb +++ b/notebooks/008_deploy_project.ipynb @@ -185,7 +185,7 @@ "metadata": {}, "outputs": [], "source": [ - "from geti_sdk.prediction_visualization.visualizer import Visualizer\n", + "from geti_sdk import Visualizer\n", "\n", "visualizer = Visualizer()\n", "\n", diff --git a/notebooks/010_model_serving.ipynb b/notebooks/010_model_serving.ipynb index c04cfd86..8175bdcd 100644 --- a/notebooks/010_model_serving.ipynb +++ b/notebooks/010_model_serving.ipynb @@ -295,7 +295,7 @@ }, "outputs": [], "source": [ - "from geti_sdk.prediction_visualization.visualizer import Visualizer\n", + "from geti_sdk import Visualizer\n", "\n", "visualizer = Visualizer()\n", "result = visualizer.draw(numpy_image, prediction)\n", diff --git a/notebooks/012_post_inference_hooks.ipynb b/notebooks/012_post_inference_hooks.ipynb index c3107df3..04a0d174 100644 --- a/notebooks/012_post_inference_hooks.ipynb +++ b/notebooks/012_post_inference_hooks.ipynb @@ -102,7 +102,7 @@ "import cv2\n", "\n", "from geti_sdk.demos import EXAMPLE_IMAGE_PATH\n", - "from geti_sdk.prediction_visualization.visualizer import Visualizer\n", + "from geti_sdk import Visualizer\n", "\n", "numpy_image = cv2.imread(EXAMPLE_IMAGE_PATH)\n", "numpy_rgb = cv2.cvtColor(numpy_image, cv2.COLOR_BGR2RGB)\n", diff --git a/notebooks/use_cases/101_simulate_low_light_product_inspection.ipynb b/notebooks/use_cases/101_simulate_low_light_product_inspection.ipynb index 2e2e64fa..8a373348 100644 --- a/notebooks/use_cases/101_simulate_low_light_product_inspection.ipynb +++ b/notebooks/use_cases/101_simulate_low_light_product_inspection.ipynb @@ -124,7 +124,7 @@ "metadata": {}, "outputs": [], "source": [ - "from geti_sdk.prediction_visualization.visualizer import Visualizer\n", + "from geti_sdk import Visualizer\n", "from geti_sdk.rest_clients import AnnotationClient\n", "\n", "annotation_client = AnnotationClient(\n",