We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ad3914b commit 6044594Copy full SHA for 6044594
onnx/test/onnx/onnx_img.py
@@ -17,10 +17,10 @@ def img2vec(img):
17
img = Image.open(join(IMG_DIR, 'cat.jpg'))
18
19
img_data = transform(img)
20
- import torch
21
- print('img data size', torch.tensor(img_data).size())
22
-
23
vec = img2vec(img)
24
print('vec', vec)
25
IMG_NORM = onnx_load('ImgNorm')
26
print('norm', IMG_NORM.run(None, {'input': img_data})[0])
+
+ # from json import dumps
+ # print(dumps(img_data[0].tolist()))
0 commit comments