COCOData.infer_on_batch

COCOData.infer_on_batch(batch, model, device) → Union[List[Tensor], Tensor]

Infer on a batch of images.

previous

COCOData.get_transform_type

next

COCOData.init_cache