tomokazu
/
helloproject-ai
mirror of https://github.com/yayoimizuha/helloproject-ai.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

update
continuous-integration/drone/push Build is passing Details

This commit is contained in:
Tomokazu Katayama 2023-10-23 06:24:57 +09:00
parent 92749421ed
commit 3b7ced3eb7
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
test_script/onnx_cacher.py
View File

@ -20,6 +20,7 @@ print(__version__)
image_arr = numpy.expand_dims(numpy.array( image_arr = numpy.expand_dims(numpy.array(
Image.open(r'C:\Users\tomokazu\CLionProjects\ameba_blog_downloader\manaka_test.jpg').convert('RGB')), 0).transpose( Image.open(r'C:\Users\tomokazu\CLionProjects\ameba_blog_downloader\manaka_test.jpg').convert('RGB')), 0).transpose(
0, 3, 1, 2).astype(numpy.float32) 0, 3, 1, 2).astype(numpy.float32)
image_arr /= 255.0
print(image_arr) print(image_arr)
print(image_arr.shape) print(image_arr.shape)
res = onnx_session.run(input_feed={'input': image_arr}, output_names=["bbox", "confidence", "landmark"]) res = onnx_session.run(input_feed={'input': image_arr}, output_names=["bbox", "confidence", "landmark"])