inference-engine/ie_bridges/python/sample/benchmark_app/benchmark/utils/constants.py

   1 """
   2  Copyright (C) 2018-2019 Intel Corporation
   3
   4  Licensed under the Apache License, Version 2.0 (the "License");
   5  you may not use this file except in compliance with the License.
   6  You may obtain a copy of the License at
   7
   8       http://www.apache.org/licenses/LICENSE-2.0
   9
  10  Unless required by applicable law or agreed to in writing, software
  11  distributed under the License is distributed on an "AS IS" BASIS,
  12  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  See the License for the specific language governing permissions and
  14  limitations under the License.
  15 """
  16
  17 HELP_MESSAGES = {
  18     'HELP': "Show this help message and exit.",
  19     'IMAGE_MESSAGE': "Required. Path to a folder with images or to image files.",
  20     'MULTI_INPUT_MESSAGE': "Optional. Path to multi input file containing.",
  21     'MODEL_MESSAGE': "Required. Path to an .xml file with a trained model.",
  22     'PLUGIN_PATH_MESSAGE': "Optional. Path to a plugin folder.",
  23     'API_MESSAGE': "Optional. Enable using sync/async API. Default value is sync",
  24     'TARGET_DEVICE_MESSAGE': "Optional. Specify a target device to infer on: CPU, GPU, FPGA, HDDL or MYRIAD. "
  25                            "Use \"-d HETERO:<comma separated devices list>\" format to specify HETERO plugin. "
  26     "The application looks for a suitable plugin for the specified device.",
  27     'ITERATIONS_COUNT_MESSAGE': "Optional. Number of iterations. "
  28     "If not specified, the number of iterations is calculated depending on a device.",
  29     'INFER_REQUESTS_COUNT_MESSAGE': "Optional. Number of infer requests (default value is 2).",
  30     'INFER_NUM_THREADS_MESSAGE': "Number of threads to use for inference on the CPU "
  31                                  "(including Hetero cases).",
  32     'CUSTOM_CPU_LIBRARY_MESSAGE': "Optional. Required for CPU custom layers. "
  33                                   "Absolute path to a shared library with the kernels implementations.",
  34     'CUSTOM_GPU_LIBRARY_MESSAGE': "Optional. Required for GPU custom kernels. Absolute path to an .xml file with the "
  35                                   "kernels description.",
  36     'BATCH_SIZE_MESSAGE': "Optional. Batch size value. If not specified, the batch size value is determined from IR",
  37     'INFER_THREADS_PINNING_MESSAGE': "Optional. Enable (\"YES\" is default value) or disable (\"NO\")"
  38                                      "CPU threads pinning for CPU-involved inference."
  39 }
  40
  41 DEVICE_DURATION_IN_SECS = {
  42     "CPU": 60,
  43     "GPU": 60,
  44     "VPU": 60,
  45     "MYRIAD": 60,
  46     "FPGA": 120,
  47     "HDDL": 60,
  48     "UNKNOWN": 120
  49 }
  50
  51 IMAGE_EXTENSIONS = ['JPEG', 'JPG', 'PNG', 'BMP']
  52
  53 MYRIAD_DEVICE_NAME = "MYRIAD"
  54 CPU_DEVICE_NAME = "CPU"
  55 GPU_DEVICE_NAME = "GPU"
  56 UNKNOWN_DEVICE_TYPE = "UNKNOWN"
  57
  58 BATCH_SIZE_ELEM = 0
  59
  60 LAYOUT_TYPE = 'NCHW'
  61
  62 XML_EXTENSION = ".xml"
  63 BIN_EXTENSION = ".bin"
  64
  65 XML_EXTENSION_PATTERN = '*' + XML_EXTENSION