diff options
Diffstat (limited to 'lib/oeqa/runtime/miutils')
4 files changed, 0 insertions, 107 deletions
diff --git a/lib/oeqa/runtime/miutils/dldtutils.py b/lib/oeqa/runtime/miutils/dldtutils.py deleted file mode 100644 index 45bf2e12..00000000 --- a/lib/oeqa/runtime/miutils/dldtutils.py +++ /dev/null | |||
| @@ -1,3 +0,0 @@ | |||
| 1 | |||
| 2 | def get_testdata_config(testdata, config): | ||
| 3 | return testdata.get(config) | ||
diff --git a/lib/oeqa/runtime/miutils/tests/dldt_inference_engine_test.py b/lib/oeqa/runtime/miutils/tests/dldt_inference_engine_test.py deleted file mode 100644 index 31bfb539..00000000 --- a/lib/oeqa/runtime/miutils/tests/dldt_inference_engine_test.py +++ /dev/null | |||
| @@ -1,56 +0,0 @@ | |||
| 1 | import os | ||
| 2 | script_path = os.path.dirname(os.path.realpath(__file__)) | ||
| 3 | files_path = os.path.join(script_path, '../../files/') | ||
| 4 | |||
| 5 | class DldtInferenceEngineTest(object): | ||
| 6 | ie_input_files = {'ie_python_sample': 'classification_sample.py', | ||
| 7 | 'input': 'chicky_512.png', | ||
| 8 | 'input_download': 'https://raw.githubusercontent.com/opencv/opencv/master/samples/data/chicky_512.png', | ||
| 9 | 'model': 'squeezenet_v1.1.xml'} | ||
| 10 | |||
| 11 | def __init__(self, target, work_dir): | ||
| 12 | self.target = target | ||
| 13 | self.work_dir = work_dir | ||
| 14 | |||
| 15 | def setup(self): | ||
| 16 | self.target.run('mkdir -p %s' % self.work_dir) | ||
| 17 | self.target.copy_to(os.path.join(files_path, 'dldt-inference-engine', self.ie_input_files['ie_python_sample']), | ||
| 18 | self.work_dir) | ||
| 19 | python_cmd = 'from openvino.inference_engine import IENetwork, IECore; ie = IECore(); print(ie.available_devices)' | ||
| 20 | __, output = self.target.run('python3 -c "%s"' % python_cmd) | ||
| 21 | self.available_devices = output | ||
| 22 | |||
| 23 | def tear_down(self): | ||
| 24 | self.target.run('rm -rf %s' % self.work_dir) | ||
| 25 | |||
| 26 | def test_check_if_openvino_device_available(self, device): | ||
| 27 | if device not in self.available_devices: | ||
| 28 | return False, self.available_devices | ||
| 29 | return True, self.available_devices | ||
| 30 | |||
| 31 | def test_can_download_input_file(self, proxy_port): | ||
| 32 | return self.target.run('cd %s; wget %s -e https_proxy=%s' % | ||
| 33 | (self.work_dir, | ||
| 34 | self.ie_input_files['input_download'], | ||
| 35 | proxy_port)) | ||
| 36 | |||
| 37 | def test_dldt_ie_classification_with_device(self, device, ir_files_dir): | ||
| 38 | return self.target.run('classification_sample_async -d %s -i %s -m %s' % | ||
| 39 | (device, | ||
| 40 | os.path.join(self.work_dir, self.ie_input_files['input']), | ||
| 41 | os.path.join(ir_files_dir, self.ie_input_files['model']))) | ||
| 42 | |||
| 43 | def test_dldt_ie_classification_python_api_with_device(self, device, ir_files_dir, extension=''): | ||
| 44 | if extension: | ||
| 45 | return self.target.run('python3 %s -d %s -i %s -m %s -l %s' % | ||
| 46 | (os.path.join(self.work_dir, self.ie_input_files['ie_python_sample']), | ||
| 47 | device, | ||
| 48 | os.path.join(self.work_dir, self.ie_input_files['input']), | ||
| 49 | os.path.join(ir_files_dir, self.ie_input_files['model']), | ||
| 50 | extension)) | ||
| 51 | else: | ||
| 52 | return self.target.run('python3 %s -d %s -i %s -m %s' % | ||
| 53 | (os.path.join(self.work_dir, self.ie_input_files['ie_python_sample']), | ||
| 54 | device, | ||
| 55 | os.path.join(self.work_dir, self.ie_input_files['input']), | ||
| 56 | os.path.join(ir_files_dir, self.ie_input_files['model']))) | ||
diff --git a/lib/oeqa/runtime/miutils/tests/dldt_model_optimizer_test.py b/lib/oeqa/runtime/miutils/tests/dldt_model_optimizer_test.py deleted file mode 100644 index 7d3db15b..00000000 --- a/lib/oeqa/runtime/miutils/tests/dldt_model_optimizer_test.py +++ /dev/null | |||
| @@ -1,23 +0,0 @@ | |||
| 1 | import os | ||
| 2 | |||
| 3 | class DldtModelOptimizerTest(object): | ||
| 4 | mo_input_files = {'model': 'squeezenet_v1.1.caffemodel', | ||
| 5 | 'prototxt': 'deploy.prototxt'} | ||
| 6 | mo_exe = 'mo.py' | ||
| 7 | |||
| 8 | def __init__(self, target, work_dir): | ||
| 9 | self.target = target | ||
| 10 | self.work_dir = work_dir | ||
| 11 | |||
| 12 | def setup(self): | ||
| 13 | self.target.run('mkdir -p %s' % self.work_dir) | ||
| 14 | |||
| 15 | def tear_down(self): | ||
| 16 | self.target.run('rm -rf %s' % self.work_dir) | ||
| 17 | |||
| 18 | def test_dldt_mo_can_create_ir(self, mo_exe_dir, mo_files_dir): | ||
| 19 | return self.target.run('python3 %s --input_model %s --input_proto %s --output_dir %s --data_type FP16' % | ||
| 20 | (os.path.join(mo_exe_dir, self.mo_exe), | ||
| 21 | os.path.join(mo_files_dir, self.mo_input_files['model']), | ||
| 22 | os.path.join(mo_files_dir, self.mo_input_files['prototxt']), | ||
| 23 | self.work_dir)) | ||
diff --git a/lib/oeqa/runtime/miutils/tests/squeezenet_model_download_test.py b/lib/oeqa/runtime/miutils/tests/squeezenet_model_download_test.py deleted file mode 100644 index a3e46a0a..00000000 --- a/lib/oeqa/runtime/miutils/tests/squeezenet_model_download_test.py +++ /dev/null | |||
| @@ -1,25 +0,0 @@ | |||
| 1 | class SqueezenetModelDownloadTest(object): | ||
| 2 | download_files = {'squeezenet1.1.prototxt': 'https://raw.githubusercontent.com/DeepScale/SqueezeNet/a47b6f13d30985279789d08053d37013d67d131b/SqueezeNet_v1.1/deploy.prototxt', | ||
| 3 | 'squeezenet1.1.caffemodel': 'https://github.com/DeepScale/SqueezeNet/raw/a47b6f13d30985279789d08053d37013d67d131b/SqueezeNet_v1.1/squeezenet_v1.1.caffemodel'} | ||
| 4 | |||
| 5 | def __init__(self, target, work_dir): | ||
| 6 | self.target = target | ||
| 7 | self.work_dir = work_dir | ||
| 8 | |||
| 9 | def setup(self): | ||
| 10 | self.target.run('mkdir -p %s' % self.work_dir) | ||
| 11 | |||
| 12 | def tear_down(self): | ||
| 13 | self.target.run('rm -rf %s' % self.work_dir) | ||
| 14 | |||
| 15 | def test_can_download_squeezenet_model(self, proxy_port): | ||
| 16 | return self.target.run('cd %s; wget %s -e https_proxy=%s' % | ||
| 17 | (self.work_dir, | ||
| 18 | self.download_files['squeezenet1.1.caffemodel'], | ||
| 19 | proxy_port)) | ||
| 20 | |||
| 21 | def test_can_download_squeezenet_prototxt(self, proxy_port): | ||
| 22 | return self.target.run('cd %s; wget %s -e https_proxy=%s' % | ||
| 23 | (self.work_dir, | ||
| 24 | self.download_files['squeezenet1.1.prototxt'], | ||
| 25 | proxy_port)) | ||
