Skip to content

Commit de7db0d

Browse files
Merge branch 'master' into ak/yolo_models_github
2 parents 4e1e113 + 7fabfbb commit de7db0d

File tree

61 files changed

+538
-378
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

61 files changed

+538
-378
lines changed

CONTRIBUTING.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,7 @@
11
# How to Contribute Models to Open Model Zoo
22

3+
> Open Model Zoo is deprecated as a source of public models. Consider adding new models to https://github.com/openvinotoolkit/openvino_notebooks
4+
35
We appreciate your intention to contribute model to the OpenVINO™ Open Model Zoo (OMZ). OMZ is licensed under the Apache\* License, Version 2.0. By contributing to the project, you agree to the license and copyright terms therein and release your contribution under these terms. Note that we accept models under permissive licenses, such as **MIT**, **Apache 2.0**, and **BSD-3-Clause**. Otherwise, it might take longer time to get your model approved.
46

57
Frameworks supported by the Open Model Zoo:

data/dataset_definitions.yml

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1462,6 +1462,17 @@ datasets:
14621462
input_suffix: in
14631463
reference_suffix: out
14641464

1465+
- name: online_mstcn_plus_encoder_dataset
1466+
data_source: annotation
1467+
reader: numpy_reader
1468+
annotation_conversion:
1469+
converter: feature_regression
1470+
reference_dir: image
1471+
input_dir: image
1472+
input_suffix: in
1473+
reference_suffix: out
1474+
use_bin_data: True
1475+
14651476
- name: smartlab_detection_10cl_top
14661477
data_source: object_detection/streams_1/top/images
14671478
annotation_conversion:

demos/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -295,7 +295,7 @@ python -mpip install --user -r <omz_dir>/demos/requirements.txt
295295
296296
### <a name="python_model_api"></a>Python\* model API package
297297
298-
To run Python demo applications, you need to install the Python* Model API package. Refer to the [Python Model API documentation](https://github.com/openvinotoolkit/open_model_zoo/blob/master/demos/common/python/openvino/model_zoo/model_api/README.md#installing-python-model-api-package)) to learn about its installation.
298+
Python* ModelAPI is factored out as a sepparate package. Refer to the [Python Model API documentation](https://github.com/openvinotoolkit/open_model_zoo/blob/master/demos/common/python/openvino/model_zoo/model_api/README.md#installing-python-model-api-package)) to learn about its installation. At the same time demos can find this package on their own. It's not required to install ModelAPI for demos.
299299
300300
### <a name="build_python_extensions"></a>Build the Native Python\* Extension Modules
301301

demos/background_subtraction_demo/python/background_subtraction_demo.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -25,15 +25,16 @@
2525
import numpy as np
2626

2727
sys.path.append(str(Path(__file__).resolve().parents[2] / 'common/python'))
28+
sys.path.append(str(Path(__file__).resolve().parents[2] / 'common/python/openvino/model_zoo'))
2829

29-
from openvino.model_zoo.model_api.models import (
30+
from model_api.models import (
3031
MaskRCNNModel, OutputTransform, RESIZE_TYPES, YolactModel,
3132
ImageMattingWithBackground, VideoBackgroundMatting, PortraitBackgroundMatting
3233
)
33-
from openvino.model_zoo.model_api.models.utils import load_labels
34-
from openvino.model_zoo.model_api.performance_metrics import PerformanceMetrics
35-
from openvino.model_zoo.model_api.pipelines import get_user_config, AsyncPipeline
36-
from openvino.model_zoo.model_api.adapters import create_core, OpenvinoAdapter, OVMSAdapter
34+
from model_api.models.utils import load_labels
35+
from model_api.performance_metrics import PerformanceMetrics
36+
from model_api.pipelines import get_user_config, AsyncPipeline
37+
from model_api.adapters import create_core, OpenvinoAdapter, OVMSAdapter
3738

3839
import monitors
3940
from images_capture import open_images_capture

demos/bert_named_entity_recognition_demo/python/bert_named_entity_recognition_demo.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -24,13 +24,14 @@
2424
from time import perf_counter
2525

2626
sys.path.append(str(Path(__file__).resolve().parents[2] / 'common/python'))
27+
sys.path.append(str(Path(__file__).resolve().parents[2] / 'common/python/openvino/model_zoo'))
2728

2829
from html_reader import get_paragraphs
2930

30-
from openvino.model_zoo.model_api.models import BertNamedEntityRecognition
31-
from openvino.model_zoo.model_api.models.tokens_bert import text_to_tokens, load_vocab_file
32-
from openvino.model_zoo.model_api.pipelines import get_user_config, AsyncPipeline
33-
from openvino.model_zoo.model_api.adapters import create_core, OpenvinoAdapter, OVMSAdapter
31+
from model_api.models import BertNamedEntityRecognition
32+
from model_api.models.tokens_bert import text_to_tokens, load_vocab_file
33+
from model_api.pipelines import get_user_config, AsyncPipeline
34+
from model_api.adapters import create_core, OpenvinoAdapter, OVMSAdapter
3435

3536
log.basicConfig(format='[ %(levelname)s ] %(message)s', level=log.DEBUG, stream=sys.stdout)
3637

demos/bert_question_answering_demo/python/bert_question_answering_demo.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -25,13 +25,14 @@
2525
import numpy as np
2626

2727
sys.path.append(str(Path(__file__).resolve().parents[2] / 'common/python'))
28+
sys.path.append(str(Path(__file__).resolve().parents[2] / 'common/python/openvino/model_zoo'))
2829

2930
from html_reader import get_paragraphs
3031

31-
from openvino.model_zoo.model_api.models import BertQuestionAnswering
32-
from openvino.model_zoo.model_api.models.tokens_bert import text_to_tokens, load_vocab_file, ContextWindow
33-
from openvino.model_zoo.model_api.pipelines import get_user_config, AsyncPipeline
34-
from openvino.model_zoo.model_api.adapters import create_core, OpenvinoAdapter, OVMSAdapter
32+
from model_api.models import BertQuestionAnswering
33+
from model_api.models.tokens_bert import text_to_tokens, load_vocab_file, ContextWindow
34+
from model_api.pipelines import get_user_config, AsyncPipeline
35+
from model_api.adapters import create_core, OpenvinoAdapter, OVMSAdapter
3536

3637
log.basicConfig(format='[ %(levelname)s ] %(message)s', level=log.DEBUG, stream=sys.stdout)
3738

demos/bert_question_answering_embedding_demo/python/bert_question_answering_embedding_demo.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -25,13 +25,14 @@
2525
import numpy as np
2626

2727
sys.path.append(str(Path(__file__).resolve().parents[2] / 'common/python'))
28+
sys.path.append(str(Path(__file__).resolve().parents[2] / 'common/python/openvino/model_zoo'))
2829

2930
from html_reader import get_paragraphs
3031

31-
from openvino.model_zoo.model_api.models import BertEmbedding, BertQuestionAnswering
32-
from openvino.model_zoo.model_api.models.tokens_bert import text_to_tokens, load_vocab_file, ContextWindow
33-
from openvino.model_zoo.model_api.pipelines import get_user_config, AsyncPipeline
34-
from openvino.model_zoo.model_api.adapters import create_core, OpenvinoAdapter
32+
from model_api.models import BertEmbedding, BertQuestionAnswering
33+
from model_api.models.tokens_bert import text_to_tokens, load_vocab_file, ContextWindow
34+
from model_api.pipelines import get_user_config, AsyncPipeline
35+
from model_api.adapters import create_core, OpenvinoAdapter
3536

3637
log.basicConfig(format='[ %(levelname)s ] %(message)s', level=log.DEBUG, stream=sys.stdout)
3738

demos/classification_demo/python/classification_demo.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -24,11 +24,12 @@
2424
import cv2
2525

2626
sys.path.append(str(Path(__file__).resolve().parents[2] / 'common/python'))
27+
sys.path.append(str(Path(__file__).resolve().parents[2] / 'common/python/openvino/model_zoo'))
2728

28-
from openvino.model_zoo.model_api.models import Classification, OutputTransform
29-
from openvino.model_zoo.model_api.performance_metrics import put_highlighted_text, PerformanceMetrics
30-
from openvino.model_zoo.model_api.pipelines import get_user_config, AsyncPipeline
31-
from openvino.model_zoo.model_api.adapters import create_core, OpenvinoAdapter, OVMSAdapter
29+
from model_api.models import Classification, OutputTransform
30+
from model_api.performance_metrics import put_highlighted_text, PerformanceMetrics
31+
from model_api.pipelines import get_user_config, AsyncPipeline
32+
from model_api.adapters import create_core, OpenvinoAdapter, OVMSAdapter
3233

3334
import monitors
3435
from images_capture import open_images_capture

demos/colorization_demo/python/colorization_demo.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,10 +25,11 @@
2525
from pathlib import Path
2626

2727
sys.path.append(str(Path(__file__).resolve().parents[2] / 'common/python'))
28+
sys.path.append(str(Path(__file__).resolve().parents[2] / 'common/python/openvino/model_zoo'))
2829

2930
import monitors
3031
from images_capture import open_images_capture
31-
from openvino.model_zoo.model_api.performance_metrics import PerformanceMetrics
32+
from model_api.performance_metrics import PerformanceMetrics
3233

3334
log.basicConfig(format='[ %(levelname)s ] %(message)s', level=log.DEBUG, stream=sys.stdout)
3435

demos/common/python/images_capture.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,14 @@
11
import os
22
import sys
3+
from pathlib import Path
34
import copy
45
from time import perf_counter
56

67
import cv2
78

9+
sys.path.append(str(Path(__file__).resolve().parents[2] / 'common/python/openvino/model_zoo'))
810

9-
from openvino.model_zoo.model_api.performance_metrics import PerformanceMetrics
11+
from model_api.performance_metrics import PerformanceMetrics
1012

1113

1214
class InvalidInput(Exception):

0 commit comments

Comments
 (0)