How to infer in batches

DefTruth / torchlm

💎A high level pipeline for face landmarks detection, it supports training, evaluating, exporting, inference(Python/C++) and 100+ data augmentations, can easily install via pip.

https://github.com/DefTruth/torchlm

MIT License

244 stars 25 forks source link

How to infer in batches #73

Open manadopeee opened 6 months ago

manadopeee commented 6 months ago

image = cv2.imread(img_path) landmarks, bboxes = torchlm.runtime.forward(image) >> batches

Is there a way to infer in batches using an image list or data loader instead of inferring with a single image?