MoGe icon indicating copy to clipboard operation
MoGe copied to clipboard

batch inference?

Open thucz opened this issue 1 year ago • 1 comments

Hi! I want to inference the monocular depths of multiple images in one batch. Do you have any idea how to make it?

thucz avatar Dec 14 '24 08:12 thucz

Hi! The model.infer() function supports both single-image inputs of shape (3, H, W) and batches of images (B, 3, H, W). The optional fov_x argument also supports both scalar and vector inputs for batched images. Please note that all images in the batch must have the same size. If the images have different sizes, I recommend multithreading for parallel inference.

EasternJournalist avatar Feb 22 '25 13:02 EasternJournalist