diff --git a/redis_consumer/consumers/mesmer_consumer.py b/redis_consumer/consumers/mesmer_consumer.py index 4112545a..144d086d 100644 --- a/redis_consumer/consumers/mesmer_consumer.py +++ b/redis_consumer/consumers/mesmer_consumer.py @@ -109,6 +109,12 @@ def _consume(self, redis_hash): scale = hvals.get('scale', '') scale = self.get_image_scale(scale, image, redis_hash) + # detect dimension order and add to redis + dim_order = self.detect_dimension_order(image, model_name, model_version) + self.update_key(redis_hash, { + 'dim_order': ','.join(dim_order) + }) + # Validate input image if hvals.get('channels'): channels = [int(c) for c in hvals.get('channels').split(',')] diff --git a/redis_consumer/consumers/mesmer_consumer_test.py b/redis_consumer/consumers/mesmer_consumer_test.py index 97c5955c..b8ba8c07 100644 --- a/redis_consumer/consumers/mesmer_consumer_test.py +++ b/redis_consumer/consumers/mesmer_consumer_test.py @@ -112,6 +112,7 @@ def test__consume(self, mocker, redis_client): mocker.patch.object(consumer, 'get_grpc_app', lambda *x, **_: mock_app) mocker.patch.object(consumer, 'get_image_scale', lambda *x, **_: 1) mocker.patch.object(consumer, 'validate_model_input', lambda *x, **_: x[0]) + mocker.patch.object(consumer, 'detect_dimension_order', lambda *x, **_: 'YXC') test_hash = 'some hash' diff --git a/redis_consumer/consumers/segmentation_consumer_test.py b/redis_consumer/consumers/segmentation_consumer_test.py index 65346983..a3f41d98 100644 --- a/redis_consumer/consumers/segmentation_consumer_test.py +++ b/redis_consumer/consumers/segmentation_consumer_test.py @@ -142,6 +142,7 @@ def test__consume(self, mocker, redis_client): mocker.patch.object(consumer, 'get_image_scale', lambda *x, **_: 1) mocker.patch.object(consumer, 'get_image_label', lambda *x, **_: 1) mocker.patch.object(consumer, 'validate_model_input', lambda *x, **_: True) + mocker.patch.object(consumer, 'detect_dimension_order', lambda *x, **_: 'YXC') test_hash = 'some hash'