IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Temporary revert test layout (#9242)

Browse Source 
* Update test_image_loading.py

* Update test_image_loading.py
This commit is contained in:
Indira Salyahova
2021-12-16 08:51:24 +03:00
committed by GitHub
parent 9b71a5fb70
commit ea3f34c351
1 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tools/pot/tests/test_image_loading.py
View File

@@ -46,15 +46,17 @@ def test_check_image(tmp_path, models, model_name, model_framework):
assert num_images_from_data_loader == num_images_in_dir
TEST_MODELS_LAYOUT = [('mobilenet-v2-pytorch', 'pytorch', 'NCHW', (3, 224, 224)),
('mobilenet-v2-pytorch', 'pytorch', 'NHWC', (224, 224, 3)),
('mobilenet-v2-pytorch', 'pytorch', None, (3, 224, 224)),
('mobilenet-v1-1.0-224-tf', 'tf', None, (224, 224, 3))]
TEST_MODELS_LAYOUT = [
#('mobilenet-v2-pytorch', 'pytorch', 'NCHW', (3, 224, 224)),
#('mobilenet-v2-pytorch', 'pytorch', 'NHWC', (224, 224, 3)),
#('mobilenet-v2-pytorch', 'pytorch', None, (3, 224, 224)),
#('mobilenet-v1-1.0-224-tf', 'tf', None, (224, 224, 3))
]
@pytest.mark.parametrize(
'model_name, model_framework, layout, reference_shape', TEST_MODELS,
ids=['{}_{}'.format(m[0], m[1]) for m in TEST_MODELS])
'model_name, model_framework, layout, reference_shape', TEST_MODELS_LAYOUT,
ids=['{}_{}_{}_{}'.format(m[0], m[1], m[2], m[3]) for m in TEST_MODELS_LAYOUT])
def test_check_layout(tmp_path, models, model_name, model_framework, layout, reference_shape):
test_dir = Path(__file__).parent
path_image_data = os.path.join(test_dir, "data/image_data")