Segment-Anything-CPP icon indicating copy to clipboard operation
Segment-Anything-CPP copied to clipboard

楼主,请问有导出模型的相关代码不?

Open TinyQi opened this issue 2 years ago • 3 comments
trafficstars

如题如图,需要使用低版本onnx导出模型,拜托楼主啦。

TinyQi avatar Aug 24 '23 12:08 TinyQi

如题如图,需要使用低版本onnx导出模型,拜托楼主啦。

等我整理一下,尽快发给你

AIDajiangtang avatar Aug 24 '23 12:08 AIDajiangtang

如题如图,需要使用低版本onnx导出模型,拜托楼主啦。

等我整理一下,尽快发给你

万分感谢啦,我已经跑通CPU了,但是由于我的生产环境的驱动版本很低,所以必须降版本。

TinyQi avatar Aug 24 '23 12:08 TinyQi

要使用低版本onnx导出模型,拜托楼主啦。 我现在发布的onnx模型就是用下面这个github导出的。 https://github.com/visheratin/segment-anything/blob/main/scripts/export_onnx_model.py

python scripts/export_onnx_model.py --checkpoint <path/to/checkpoint> --model-type <model_type> --output <path/to/output> 可以加一个--opset参数指定onnx版本。 如果想输入动态的图像大小,或者动态batch大小可以在上述代码中修改dynamic_axes,但我没试过哦

AIDajiangtang avatar Aug 25 '23 02:08 AIDajiangtang