Relevant code for batching of images (we call this function in tasks/formula_recognition.py): https://github.com/opendatalab/UniMERNet/blob/038f6dd26062dc0ae0b85915286fb3ca31cdd96e/unimernet/models/unimernet/unimernet.py#L68
Relevant code for batching of images (we call this function in tasks/formula_recognition.py):
https://github.com/opendatalab/UniMERNet/blob/038f6dd26062dc0ae0b85915286fb3ca31cdd96e/unimernet/models/unimernet/unimernet.py#L68