amd
/

mobilenet_v2_1.0_224

Image Classification

Model card Files Files and versions Community

haoliangtan commited on Jan 8

Commit

25009be

•

1 Parent(s): 6f8454f

Update eval_onnx.py

Files changed (1) hide show

eval_onnx.py +1 -0

eval_onnx.py CHANGED Viewed

@@ -141,6 +141,7 @@ def val_imagenet():
   val_loader = tqdm(val_loader, file=sys.stdout)
   with torch.no_grad():
     for batch_idx, (images, targets) in enumerate(val_loader):
       inputs, targets = images.numpy(), targets
       ort_inputs = {ort_session.get_inputs()[0].name: inputs}

   val_loader = tqdm(val_loader, file=sys.stdout)
   with torch.no_grad():
     for batch_idx, (images, targets) in enumerate(val_loader):
+      images = torch.permute(images, (0, 2, 3, 1))
       inputs, targets = images.numpy(), targets
       ort_inputs = {ort_session.get_inputs()[0].name: inputs}