search

developmentseed / segment-anything-services

Running segment-anything image embedding, prompting, and mask generation as torchserve services
Apache License 2.0
92 stars 10 forks source link

compile the encoder model for the GPU or CPU #5

Closed rbavery closed 9 months ago

rbavery commented 1 year ago

The examples in the original repo show how to compile the decoder model but not the encoder model.

this PR below shows how to compile the encoder model as well for onnx

https://github.com/facebookresearch/segment-anything/pull/29/files