mirror of
git://git.yoctoproject.org/meta-intel.git
synced 2025-07-19 21:09:03 +02:00
1e514838df
Add sanity tests for inference engine:
- test inference engine c/cpp shared library
- test inference engine python api
- test inference engine cpu, gpu, myriad plugin
Add sanity tests for model optimizer
- test model optmizer can generate ir
Licenses:
- classification_sample.py
license: Apache 2.0
source: <install_root>/deployment_tools/inference_engine/samples/*
Signed-off-by: Yeoh Ee Peng <ee.peng.yeoh@intel.com>
Signed-off-by: Anuj Mittal <anuj.mittal@intel.com>
49 lines
2.5 KiB
Python
49 lines
2.5 KiB
Python
import os
|
|
script_path = os.path.dirname(os.path.realpath(__file__))
|
|
files_path = os.path.join(script_path, '../../files/')
|
|
|
|
class DldtInferenceEngineTest(object):
|
|
ie_input_files = {'ie_python_sample': 'classification_sample.py',
|
|
'input': 'chicky_512.png',
|
|
'input_download': 'https://raw.githubusercontent.com/opencv/opencv/master/samples/data/chicky_512.png',
|
|
'model': 'squeezenet_v1.1.xml'}
|
|
|
|
def __init__(self, target, work_dir):
|
|
self.target = target
|
|
self.work_dir = work_dir
|
|
|
|
def setup(self):
|
|
self.target.run('mkdir -p %s' % self.work_dir)
|
|
self.target.copy_to(os.path.join(files_path, 'dldt-inference-engine', self.ie_input_files['ie_python_sample']),
|
|
self.work_dir)
|
|
|
|
def tear_down(self):
|
|
self.target.run('rm -rf %s' % self.work_dir)
|
|
|
|
def test_can_download_input_file(self, proxy_port):
|
|
return self.target.run('cd %s; wget %s -e https_proxy=%s' %
|
|
(self.work_dir,
|
|
self.ie_input_files['input_download'],
|
|
proxy_port))
|
|
|
|
def test_dldt_ie_classification_with_device(self, device, ir_files_dir):
|
|
return self.target.run('classification_sample_async -d %s -i %s -m %s' %
|
|
(device,
|
|
os.path.join(self.work_dir, self.ie_input_files['input']),
|
|
os.path.join(ir_files_dir, self.ie_input_files['model'])))
|
|
|
|
def test_dldt_ie_classification_python_api_with_device(self, device, ir_files_dir, extension=''):
|
|
if extension:
|
|
return self.target.run('python3 %s -d %s -i %s -m %s -l %s' %
|
|
(os.path.join(self.work_dir, self.ie_input_files['ie_python_sample']),
|
|
device,
|
|
os.path.join(self.work_dir, self.ie_input_files['input']),
|
|
os.path.join(ir_files_dir, self.ie_input_files['model']),
|
|
extension))
|
|
else:
|
|
return self.target.run('python3 %s -d %s -i %s -m %s' %
|
|
(os.path.join(self.work_dir, self.ie_input_files['ie_python_sample']),
|
|
device,
|
|
os.path.join(self.work_dir, self.ie_input_files['input']),
|
|
os.path.join(ir_files_dir, self.ie_input_files['model'])))
|