deepinsight / deepinsight/insightface

arcface r50 onnx2caffe conversion

Open
#1,614 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
29.7k
Forks
6.1k
PR merge metrics
No merged PRs in 30d

Description

When converting the r50 model trained from "arcface-pytorch" to the caffe model, there is an error.

Which part should I check during the torch -> onnx -> caffe converting process?

```
./onnx2caffe.sh: line 1: 52722 Floating point exception(core dumped) python3 convertCaffe.py ./backbone.onnx ./arcface_r50.prototxt ./arcface_r50.caffemodel
```

_operators.py
```
opset_version = 11

File "/home_sw1/phj8498/insightface/tools/onnx2caffe/onnx2caffe/_operators.py", line 33, in make_input
"ONNX opset version {} is not supported,only opset 9 is supported,you can export onnx by setting opset_version like torch.onnx.export(model, '', opset_version=9, verbose=True).\n".format(version )
TypeError: ONNX opset version 11 is not supported,only opset 9 is supported,you can export onnx by setting opset_version like torch.onnx.export(model, '', opset_version=9, verbose=True).
```

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.