djl
djl copied to clipboard
Request for SAM2 Model in ONNX Format and Synset File for Classification Output
Description
Hello DJL Team,
I am currently using the SAM2 model from the DJL ModelZoo for inference purposes. However, I have encountered a couple of limitations that I would like to request your support in resolving:
ONNX Model Format: Currently, the SAM2 model is only available in the PyTorch (.pt) format. For broader compatibility and ease of integration with other frameworks, I would like to request that an ONNX (.onnx) format version of the SAM2 model be added to the ModelZoo.
Classification Output and Synset File: It appears that the SAM2 model's inference results do not provide classification outputs. To facilitate classification tasks, could you please provide a synset.txt file or any relevant label mapping file for this model, so that users can easily interpret the classification results?
These features would greatly enhance the usability and flexibility of the SAM2 model in a variety of projects.
Thank you for your work and consideration!