ZF/insightface
1
0
Fork 0
mirror of https://github.com/deepinsight/insightface.git synced 2025-12-30 08:02:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update torch2onnx.py

Browse Source 
For ViT
This commit is contained in:
Xiang An
2022-05-03 14:10:09 +08:00
committed by GitHub
parent 92c6bd9a4e
commit b5958bc1d8
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
recognition/arcface_torch/torch2onnx.py
View File

@@ -3,7 +3,7 @@ import onnx
import torch
def convert_onnx(net, path_module, output, opset=9, simplify=False):
def convert_onnx(net, path_module, output, opset=11, simplify=False):
assert isinstance(net, torch.nn.Module)
img = np.random.randint(0, 255, size=(112, 112, 3), dtype=np.int32)
img = img.astype(np.float)