Inference using multigpu on lots of images.

Media-Smart / vedadet

A single stage object detection toolbox based on PyTorch

Apache License 2.0

498 stars 128 forks source link

Inference using multigpu on lots of images. #42

Closed MagicFrogSJTU closed 3 years ago

MagicFrogSJTU commented 3 years ago

The tool/infer.py is designed to process only an image at a time with one gpu. What if there are lots of images to infer? Is there any tutorials or small code snipets to implement large-scale inference?

hxcai commented 3 years ago

@MagicFrogSJTU You mean process batch images? You can modify the infer code, please refer to collate and scatter implementation detail in infer code.