amd
/

mobilenet_v2_1.0_224

Image Classification

Model card Files Files and versions Community

Update README.md

#4

by haoliangtan - opened Jan 12

base: refs/heads/main

←

from: refs/pr/4

Discussion Files changed

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -63,7 +63,7 @@ You can use the raw model for image classification. See the [model hub](https://
                   transforms.ToTensor(),
                   normalize])
   img_tensor = img_transformer(Image.open(image_path)).unsqueeze(0)
   so = onnxruntime.SessionOptions()
   ort_session = onnxruntime.InferenceSession(
           onnx_model, so,

                   transforms.ToTensor(),
                   normalize])
   img_tensor = img_transformer(Image.open(image_path)).unsqueeze(0)
+  img_tensor = torch.permute(img_tensor, (0, 2, 3, 1))
   so = onnxruntime.SessionOptions()
   ort_session = onnxruntime.InferenceSession(
           onnx_model, so,