GPU version on the way？

dinglufe / segment-anything-cpp-wrapper

MIT License

223 stars 35 forks source link

Closed julinfn closed 1 year ago

dinglufe commented 1 year ago

Yes, I am still working on it.

dinglufe commented 1 year ago

Release 1.3 now supports execution using CUDA.

bnascimento commented 1 year ago

do we need to re-export the onnx models to CUDA? how?

dinglufe commented 1 year ago

No need to re-export. But using CUDA did not provide a significant improvement in efficiency (only about 4 times faster on my computer).