amd
/

mobilenet_v2_1.0_224

Image Classification

Model card Files Files and versions

haoliangtan commited on Jan 12, 2024

Commit

8603bc2

·

verified ·

1 Parent(s): 162de94

Update README.md

change NCHW to NHWC

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,