The running result is all black（img_demo.png）

[Snaillms]

Dear author, I am very interested in your work, but I have a problem. I try to run MedSAM_Inference.py, in which the pre-trained weights and test images (assets/img_demo.png) have been changed to the correct path. Why is the result after running it completely black?
I did not change the parameters of the original box. Of course, I tried other box parameters and the result was also completely black.
Looking forward to your reply

[MaeenBadea]

just add a small change to scale the image in the visible range 0-255

io.imsave(
    join(args.seg_path, "seg_" + os.path.basename(args.data_path)),
    (medsam_seg * 255).astype(np.uint8),  # <- scale binary mask to visible range
    check_contrast=False,
)