Segment-Anything-CPP
Segment-Anything-CPP copied to clipboard
楼主,请问有导出模型的相关代码不?
trafficstars
如题如图,需要使用低版本onnx导出模型,拜托楼主啦。
如题如图,需要使用低版本onnx导出模型,拜托楼主啦。
等我整理一下,尽快发给你
如题如图,需要使用低版本onnx导出模型,拜托楼主啦。
等我整理一下,尽快发给你
万分感谢啦,我已经跑通CPU了,但是由于我的生产环境的驱动版本很低,所以必须降版本。
要使用低版本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,但我没试过哦