From 91506d29cf21cf91a0aeb790a55da574dd03f024 Mon Sep 17 00:00:00 2001 From: "Tybulewicz, Tomasz" Date: Wed, 27 Aug 2025 10:04:29 +0200 Subject: [PATCH 1/3] Models from geti_predict used in tests --- tests/accuracy/prepare_data.py | 6 +++--- tests/precommit/prepare_data.py | 6 +++--- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/accuracy/prepare_data.py b/tests/accuracy/prepare_data.py index 6b4da535..5eff8487 100644 --- a/tests/accuracy/prepare_data.py +++ b/tests/accuracy/prepare_data.py @@ -35,19 +35,19 @@ async def download_otx_model(client, otx_models_dir, model_name, format="xml"): if format == "onnx": await stream_file( client, - f"https://storage.openvinotoolkit.org/repositories/model_api/test/otx_models/{model_name}/model.onnx", + f"https://storage.geti.intel.com/geti_predict/test/otx_models/{model_name}/model.onnx", f"{otx_models_dir}/{model_name}.onnx", ) else: await asyncio.gather( stream_file( client, - f"https://storage.openvinotoolkit.org/repositories/model_api/test/otx_models/{model_name}/openvino.xml", + f"https://storage.geti.intel.com/geti_predict/test/otx_models/{model_name}/openvino.xml", f"{otx_models_dir}/{model_name}.xml", ), stream_file( client, - f"https://storage.openvinotoolkit.org/repositories/model_api/test/otx_models/{model_name}/openvino.bin", + f"https://storage.geti.intel.com/geti_predict/test/otx_models/{model_name}/openvino.bin", f"{otx_models_dir}/{model_name}.bin", ), ) diff --git a/tests/precommit/prepare_data.py b/tests/precommit/prepare_data.py index b0e78966..a08b75cf 100644 --- a/tests/precommit/prepare_data.py +++ b/tests/precommit/prepare_data.py @@ -14,16 +14,16 @@ def retrieve_otx_model(data_dir, model_name, format="xml"): destination_folder.mkdir(parents=True, exist_ok=True) if format == "onnx": urlretrieve( - f"https://storage.openvinotoolkit.org/repositories/model_api/test/otx_models/{model_name}/model.onnx", + f"https://storage.geti.intel.com/geti_predict/test/otx_models/{model_name}/model.onnx", destination_folder / f"{model_name}.onnx", ) else: urlretrieve( - f"https://storage.openvinotoolkit.org/repositories/model_api/test/otx_models/{model_name}/openvino.xml", + f"https://storage.geti.intel.com/geti_predict/test/otx_models/{model_name}/openvino.xml", destination_folder / f"{model_name}.xml", ) urlretrieve( - f"https://storage.openvinotoolkit.org/repositories/model_api/test/otx_models/{model_name}/openvino.bin", + f"https://storage.geti.intel.com/geti_predict/test/otx_models/{model_name}/openvino.bin", f"{destination_folder}/{model_name}.bin", ) From 6a54c99883a28b142c1e2c7e19122e54114defe8 Mon Sep 17 00:00:00 2001 From: "Tybulewicz, Tomasz" Date: Wed, 27 Aug 2025 11:33:09 +0200 Subject: [PATCH 2/3] Add missing tinynet_imagenet model to accuracy tests --- tests/accuracy/public_scope.json | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/tests/accuracy/public_scope.json b/tests/accuracy/public_scope.json index 3ff8d0e1..d61fb98b 100644 --- a/tests/accuracy/public_scope.json +++ b/tests/accuracy/public_scope.json @@ -207,6 +207,16 @@ } ] }, + { + "name": "otx_models/tinynet_imagenet.xml", + "type": "ClassificationModel", + "test_data": [ + { + "image": "coco128/images/train2017/000000000471.jpg", + "reference": ["42 (42): 1.000, [0], [0], [0]"] + } + ] + }, { "name": "otx_models/is_efficientnetb2b_maskrcnn_coco_reduced.xml", "type": "MaskRCNNModel", From d9eed3ffb9c818b9ad04549430a561d58aa02b9c Mon Sep 17 00:00:00 2001 From: "Tybulewicz, Tomasz" Date: Wed, 27 Aug 2025 12:33:16 +0200 Subject: [PATCH 3/3] Download tinynet_imagenet dataset for accuracy tests --- tests/accuracy/prepare_data.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/accuracy/prepare_data.py b/tests/accuracy/prepare_data.py index 5eff8487..622e08bc 100644 --- a/tests/accuracy/prepare_data.py +++ b/tests/accuracy/prepare_data.py @@ -88,6 +88,7 @@ async def main(): ), download_otx_model(client, otx_models_dir, "cls_efficient_b0_cars"), download_otx_model(client, otx_models_dir, "cls_efficient_v2s_cars"), + download_otx_model(client, otx_models_dir, "tinynet_imagenet"), download_otx_model(client, otx_models_dir, "Lite-hrnet-18"), download_otx_model(client, otx_models_dir, "Lite-hrnet-18_mod2"), download_otx_model(client, otx_models_dir, "Lite-hrnet-s_mod2"),