Pillow modes in the preprocessor's resize mechanism (#16601)

This commit is contained in:
Tomasz Dołbniak
2023-04-12 15:30:42 +02:00
committed by GitHub
parent ed5148b75f
commit fb49228fec
6 changed files with 126 additions and 38 deletions

View File

@@ -393,7 +393,7 @@ def test_graph_preprocess_steps(algorithm, color_format1, color_format2, is_fail
"Gather",
"Interpolate",
]
assert len(model_operators) == 16
assert len(model_operators) == 15
assert function.get_output_size() == 1
assert list(function.get_output_shape(0)) == [1, 3, 3, 3]
assert function.get_output_element_type(0) == Type.f32
@@ -535,7 +535,7 @@ def test_graph_preprocess_resize_algorithm():
"Subtract",
"Interpolate",
]
assert len(model_operators) == 8
assert len(model_operators) == 7
assert function.get_output_size() == 1
assert list(function.get_output_shape(0)) == [1, 1, 3, 3]
assert function.get_output_element_type(0) == Type.f32