colorByArtImagePipeline / inspect_sam.py
s0sp
cpu intensive without tweak best result
a9e6328
Raw
History Blame Contribute Delete
274 Bytes
# inspect_sam.py
import onnxruntime as ort
path = "models/sam2_decoder.onnx"
session = ort.InferenceSession(path)
print("SAM2 Decoder Inputs:")
for i in session.get_inputs():
print(f" Name: {i.name}")
print(f" Shape: {i.shape}")
print(f" Type: {i.type}")