def test_default_keep_aspect_ratio_resizer(self):
self.pipeline_config._model_params['resizer_min_dimension'] = 600
self.pipeline_config._model_params['resizer_max_dimension'] = 1024
self.assertTupleEqual(
(600, 600),
calculate_placeholder_spatial_shape(self.graph, self.match,
self.pipeline_config))
def test_default_fixed_shape_resizer(self):
self.pipeline_config._model_params['resizer_image_height'] = 300
self.pipeline_config._model_params['resizer_image_width'] = 600
self.assertTupleEqual(
(300, 600),
calculate_placeholder_spatial_shape(self.graph, self.match,
self.pipeline_config))
def test_keep_aspect_ratio_resizer_overrided_by_user_pad(self):
self.pipeline_config._model_params['resizer_min_dimension'] = 600
self.pipeline_config._model_params['resizer_max_dimension'] = 1024
self.pipeline_config._model_params['pad_to_max_dimension'] = True
self.graph.graph['user_shapes'] = {'image_tensor': [{'shape': [1, 400, 300, 3]}]}
self.assertTupleEqual((1024, 1024),
calculate_placeholder_spatial_shape(self.graph, self.match, self.pipeline_config))
def test_fixed_shape_resizer_overrided_by_user(self):
self.pipeline_config._model_params['resizer_image_height'] = 300
self.pipeline_config._model_params['resizer_image_width'] = 600
self.graph.graph['user_shapes'] = {
'image_tensor': [{
'shape': [1, 400, 500, 3]
}]
}
self.assertTupleEqual(
(400, 500),
calculate_placeholder_spatial_shape(self.graph, self.match,
self.pipeline_config))
