- mmdet: 因为仓库中已经有了mmdet离线包,该包是下载的2.25.0,为了可以转onnx,里面做了一些修改,不影响现在的推理
- 其他库安装:
pip install -r requirements.txt -i https://pypi.douban.com/simple/
python test_pub.py
# 输出结果,会在output目录下保存绘制框线的图像
# OK
- 状态:可以成功转换,转换时也特别耗费内存-_-!,同时因基于ONNXRuntime推理时,太耗费内存,而放弃
- 转换脚本:
- 转换之后onnx模型下载link,仅供参考,推理尚未成功
- 验证推理代码:
# 在davarocr/davar_common/apis/inference.py#L119行插入
import onnxruntime
session = onnxruntime.InferenceSession('lgpma.onnx')
input_name = session.get_inputs()[0].name
inputs = {
input_name: data['img'][0].numpy().astype(np.float32)
}
outs = session.run(None, inputs)