Inference with specified Region of Interest

Hi everyone, I was wondering if there is an option on torchio where I can do inference upon an image but with a specified region of interest.

What I am trying to say is:

1. I have: Torchio Subject, trained model, RoI mask mask.nii.gz
2. Using this mask.nii.gz, I will find a bounding box, and only forward pass the bounding boxed image through the model (with some padding if necessary - to maintain the downsampled dimensions in the U-Net)

I was wondering if there is any way where I can add the bounding box as another attribute of the torchio Subjectobject and it does inference automatically upon that.

Please let me know if something is not clear. Thanks, Megh

[meghbhalerao]

Thank you @fepegar, I think the crop/pad function would solve my problem - I can give a mask as a LabelMap and use the croporpad method to crop the image and then forward pass it. Thanks again

[meghbhalerao]

Thanks for your reply, @fepegar. I will open a new issue as this could happen for other transforms as well. Thanks, Megh