fix issue log print wrong origin image shape (#581)

2020-05-26 19:27:14 +08:00 · 2020-05-26 19:27:14 +08:00 · 7595512d1f
commit 7595512d1f
parent c3aa866a33
1 changed files with 1 additions and 1 deletions
--- a/inference-engine/ie_bridges/python/sample/object_detection_sample_ssd/object_detection_sample_ssd.py
+++ b/inference-engine/ie_bridges/python/sample/object_detection_sample_ssd/object_detection_sample_ssd.py
@ -101,8 +101,8 @@ def main():
        log.info("File was added: ")
        log.info("        {}".format(args.input[i]))
        if (ih, iw) != (h, w):
-            image = cv2.resize(image, (w, h))
            log.warning("Image {} is resized from {} to {}".format(args.input[i], image.shape[:-1], (h, w)))
+            image = cv2.resize(image, (w, h))
        image = image.transpose((2, 0, 1))  # Change data layout from HWC to CHW
        images[i] = image