recognition.py

Go to the documentation of this file.

import cv2 as cv
 
from .base_metric import BaseMetric
from ..factory import METRICS
 
@METRICS.register
class Recognition(BaseMetric):
    def __init__(self, **kwargs):
        super().__init__(**kwargs)
 
    def forward(self, model, *args, **kwargs):
        img, bboxes = args
 
        self._timer.reset()
        for idx, bbox in enumerate(bboxes):
            for _ in range(self._warmup_warmup):
                model.infer(img, bbox)
            for _ in range(self._repeat_repeat):
                self._timer.start()
                model.infer(img, bbox)
                self._timer.stop()
 
        return self._getResult()