MoGe
MoGe copied to clipboard
batch inference?
Hi! I want to inference the monocular depths of multiple images in one batch. Do you have any idea how to make it?
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.