Commit
•
b8b1ada
1
Parent(s):
162de94
Update README.md (#4)
Browse files- Update README.md (8603bc245ae768b7579ee1299672981e96e417ca)
Co-authored-by: Haoliang Tan <[email protected]>
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,
|