inference and geneal Dataset class are needed.

[MelindaDong]

Hey,

I am trying to do some inference with your code. I think it will be better if you can:

1. Add a Dataset class for a general use case, like with options like 'ct''mri''2d''3d', instead of a separate Dataset for a specific dataset.
2. Add an inference.py for directly inference with the 'MedSAM2_pretrain.pth' weight without training.

I think reproducing friendly code will help to gain more citations.

Thanks a lot.

[WendellAtHome]

Any updates on progress? It would be super helpful if you have implemented it.