18 lines
543 B
Python
18 lines
543 B
Python
# Copyright (C) 2018-2022 Intel Corporation
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
import os
|
|
|
|
|
|
if os.environ.get('OUTPUT_DIR') is not None:
|
|
out_path = os.environ['OUTPUT_DIR']
|
|
else:
|
|
script_path = os.path.dirname(os.path.realpath(__file__))
|
|
out_path = os.path.join(script_path, 'out')
|
|
if not os.path.exists(out_path):
|
|
os.makedirs(out_path)
|
|
|
|
# supported_devices : CPU, GPU, MYRIAD
|
|
test_device = os.environ.get('TEST_DEVICE', 'CPU;GPU').split(';')
|
|
test_precision = os.environ.get('TEST_PRECISION', 'FP32;FP16').split(';')
|