COCOData.infer_on_batch#

COCOData.infer_on_batch(batch, model, device) List[Tensor][source]#

Infer on a batch of images.