Upload 2 files
Browse files- ese_vovnet39b_int.onnx +2 -2
- eval_onnx.py +1 -1
ese_vovnet39b_int.onnx
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e056581c127708ee9751d6402fe7baa9c9fced577a240054e1c982d8a1df6856
|
3 |
+
size 102032283
|
eval_onnx.py
CHANGED
@@ -144,7 +144,7 @@ def val_imagenet():
|
|
144 |
val_loader = tqdm(val_loader, file=sys.stdout)
|
145 |
with torch.no_grad():
|
146 |
for batch_idx, (images, targets) in enumerate(val_loader):
|
147 |
-
inputs, targets = images.numpy(), targets
|
148 |
ort_inputs = {ort_session.get_inputs()[0].name: inputs}
|
149 |
|
150 |
outputs = ort_session.run(None, ort_inputs)
|
|
|
144 |
val_loader = tqdm(val_loader, file=sys.stdout)
|
145 |
with torch.no_grad():
|
146 |
for batch_idx, (images, targets) in enumerate(val_loader):
|
147 |
+
inputs, targets = images.permute([0,2,3,1]).numpy(), targets
|
148 |
ort_inputs = {ort_session.get_inputs()[0].name: inputs}
|
149 |
|
150 |
outputs = ort_session.run(None, ort_inputs)
|