From 5759a13077321dcd7178413267f6bc11d43cf399 Mon Sep 17 00:00:00 2001 From: wuxuejian Date: Thu, 9 Jul 2020 10:35:16 +0800 Subject: [PATCH] adapt crop and resize op check --- mindspore/ops/operations/image_ops.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mindspore/ops/operations/image_ops.py b/mindspore/ops/operations/image_ops.py index 1e366b5ea6..437cda3301 100644 --- a/mindspore/ops/operations/image_ops.py +++ b/mindspore/ops/operations/image_ops.py @@ -117,8 +117,8 @@ class CropAndResize(PrimitiveWithInfer): validator.check("crop_height", crop_size_value[0], "minimum", 0, Rel.GT, self.name) validator.check("crop_width", crop_size_value[1], "minimum", 0, Rel.GT, self.name) # check crop_size element type - validator.check("crop_height dtype", crop_size_dtype[0], mstype.int32, self.name) - validator.check("crop_width dtype", crop_size_dtype[1], mstype.int32, self.name) + validator.check("crop_height dtype", crop_size_dtype[0], "expected", mstype.int32, Rel.EQ, self.name) + validator.check("crop_width dtype", crop_size_dtype[1], "expected", mstype.int32, Rel.EQ, self.name) num_boxes = boxes_shape[0] crop_height = crop_size_value[0]