How to get inference on multiple images batchwise in one go?

google-research / pix2seq

Pix2Seq codebase: multi-tasks with generative modeling (autoregressive and diffusion)

Apache License 2.0

857 stars 71 forks source link

How to get inference on multiple images batchwise in one go? #36

Open sachin-rock-gh opened 1 year ago

sachin-rock-gh commented 1 year ago

Can someone guide how to implement inference on multiple images rather than passing one single image? In the codebase I don't see scope for multiple inference or am I missing something? Thanks.