how to prediction without using main. py of SAM-Med2D onnxruntime？how to modify test.py to predicte to get mask？

OpenGVLab / SAM-Med2D

Official implementation of SAM-Med2D

Apache License 2.0

885 stars 84 forks source link

how to prediction without using main. py of SAM-Med2D onnxruntime？how to modify test.py to predicte to get mask？ #68

Open dw1360585641 opened 4 months ago

dw1360585641 commented 4 months ago

How to change test.py to prediction, because I only want to get the segmentation mask. besides， Can we modify the main. py of SAM-Med2D onnxruntime to the predicted output mask