From 89c27fa8203b8e318c3e19cf373e1e2cc2ec243b Mon Sep 17 00:00:00 2001 From: Adrian Boguszewski Date: Mon, 20 Jan 2025 17:15:39 +0100 Subject: [PATCH] Made explainable AI consistent with the kit template (#171) * Made explainable AI consistent with the kit template * Updated requirements --- ai_ref_kits/explainable_ai/README.md | 2 +- ai_ref_kits/explainable_ai/{ => data}/Cars-FHD.mov | 0 ai_ref_kits/explainable_ai/explainable_ai.ipynb | 2 +- ai_ref_kits/explainable_ai/requirements.txt | 6 ++++-- 4 files changed, 6 insertions(+), 4 deletions(-) rename ai_ref_kits/explainable_ai/{ => data}/Cars-FHD.mov (100%) diff --git a/ai_ref_kits/explainable_ai/README.md b/ai_ref_kits/explainable_ai/README.md index af8492b5..e6e868af 100644 --- a/ai_ref_kits/explainable_ai/README.md +++ b/ai_ref_kits/explainable_ai/README.md @@ -39,7 +39,7 @@ New updates will be added here. ## Installing Prerequisites -Now, let's dive into the steps starting with installing Python. We recommend using Ubuntu to set up and run this project. This project requires Python 3.8 or higher and a few libraries. If you don't have Python installed on your machine, go to https://www.python.org/downloads/ and download the latest version for your operating system. Follow the prompts to install Python, making sure to check the option to add Python to your PATH environment variable. +Now, let's dive into the steps starting with installing Python. We recommend using Ubuntu to set up and run this project. This project requires Python 3.9 or higher and a few libraries. If you don't have Python installed on your machine, go to https://www.python.org/downloads/ and download the latest version for your operating system. Follow the prompts to install Python, making sure to check the option to add Python to your PATH environment variable. Install libraries and tools: diff --git a/ai_ref_kits/explainable_ai/Cars-FHD.mov b/ai_ref_kits/explainable_ai/data/Cars-FHD.mov similarity index 100% rename from ai_ref_kits/explainable_ai/Cars-FHD.mov rename to ai_ref_kits/explainable_ai/data/Cars-FHD.mov diff --git a/ai_ref_kits/explainable_ai/explainable_ai.ipynb b/ai_ref_kits/explainable_ai/explainable_ai.ipynb index dbfe8a71..0eda48d1 100644 --- a/ai_ref_kits/explainable_ai/explainable_ai.ipynb +++ b/ai_ref_kits/explainable_ai/explainable_ai.ipynb @@ -688,7 +688,7 @@ "import cv2\n", "\n", "model = YOLO('model/yolov8m_openvino_model/')\n", - "cap = cv2.VideoCapture(\"Cars-FHD.mov\")\n", + "cap = cv2.VideoCapture(\"data/Cars-FHD.mov\")\n", "assert cap.isOpened(), \"Error reading video file\"\n", "w, h, fps = (int(cap.get(x)) for x in (cv2.CAP_PROP_FRAME_WIDTH, cv2.CAP_PROP_FRAME_HEIGHT, cv2.CAP_PROP_FPS))\n", "\n", diff --git a/ai_ref_kits/explainable_ai/requirements.txt b/ai_ref_kits/explainable_ai/requirements.txt index f54b4f4b..5432cf54 100644 --- a/ai_ref_kits/explainable_ai/requirements.txt +++ b/ai_ref_kits/explainable_ai/requirements.txt @@ -1,13 +1,15 @@ +--extra-index-url https://download.pytorch.org/whl/cpu + openvino==2024.4.0 nncf==2.13.0 ultralytics==8.3.23 -numpy<1.27 +numpy==1.26.4 pillow==11.0.0 opencv-python==4.10.0.84 onnx==1.17.0 torch==2.5.0 torchvision==0.20.0 -supervision==0.24.0 +supervision==0.25.1 jupyterlab==4.2.5 tqdm==4.66.6 pycocotools==2.0.8