Is there any script for batch inference?

facebookresearch / Detectron

FAIR's research platform for object detection research, implementing popular algorithms like Mask R-CNN and RetinaNet.

Apache License 2.0

26.21k stars 5.45k forks source link

Open rogercmq opened 2 years ago

rogercmq commented 2 years ago

detectron/tools/infer_simple.py is a clean and tiny inference demo for the case when batchsize equals to 1.

For efficient inference, I want to detect several images at the same time.

Could anyone give me some advice on implementing "batchsize > 1" inference demo?