From 3b7ced3eb7ffbadec214e7733dae4f821debad01 Mon Sep 17 00:00:00 2001 From: Tomokazu Katayama Date: Mon, 23 Oct 2023 06:24:57 +0900 Subject: [PATCH] update --- test_script/onnx_cacher.py | 1 + 1 file changed, 1 insertion(+) diff --git a/test_script/onnx_cacher.py b/test_script/onnx_cacher.py index 3394b37..a9cc4ab 100644 --- a/test_script/onnx_cacher.py +++ b/test_script/onnx_cacher.py @@ -20,6 +20,7 @@ print(__version__) image_arr = numpy.expand_dims(numpy.array( Image.open(r'C:\Users\tomokazu\CLionProjects\ameba_blog_downloader\manaka_test.jpg').convert('RGB')), 0).transpose( 0, 3, 1, 2).astype(numpy.float32) +image_arr /= 255.0 print(image_arr) print(image_arr.shape) res = onnx_session.run(input_feed={'input': image_arr}, output_names=["bbox", "confidence", "landmark"])