zhengrongzhang haoliangtan commited on
Commit
b8b1ada
1 Parent(s): 162de94

Update README.md (#4)

Browse files

- Update README.md (8603bc245ae768b7579ee1299672981e96e417ca)


Co-authored-by: Haoliang Tan <[email protected]>

Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -63,7 +63,7 @@ You can use the raw model for image classification. See the [model hub](https://
63
  transforms.ToTensor(),
64
  normalize])
65
  img_tensor = img_transformer(Image.open(image_path)).unsqueeze(0)
66
-
67
  so = onnxruntime.SessionOptions()
68
  ort_session = onnxruntime.InferenceSession(
69
  onnx_model, so,
 
63
  transforms.ToTensor(),
64
  normalize])
65
  img_tensor = img_transformer(Image.open(image_path)).unsqueeze(0)
66
+ img_tensor = torch.permute(img_tensor, (0, 2, 3, 1))
67
  so = onnxruntime.SessionOptions()
68
  ort_session = onnxruntime.InferenceSession(
69
  onnx_model, so,