Python bgsub_gbb Examples

Example #1

File: detector.py Project: VyomUnadkat/intelligent-traffic-management-and-violation-detection-system

def get_bounding_boxes(frame, model):
    if model == 'yolo':
        return yolo_gbb(frame)
    elif model == 'haarc':
        return hc_gbb(frame)
    elif model == 'bgsub':
        return bgsub_gbb(frame)
    elif model == 'ssd':
        return ssd_gbb(frame)
    else:
        raise Exception('Invalid detector model/algorithm specified (options: yolo, haarc, bgsub, ssd)')

Example #2

File: detector.py Project: shanliwa1/Vehicle-Counting

def get_bounding_boxes(frame, model):
    if model == 'yolo':
        from detectors.yolo.yolo_detector import get_bounding_boxes as yolo_gbb
        return yolo_gbb(frame)
    elif model == 'haarc':
        from detectors.haarc.hc_detector import get_bounding_boxes as hc_gbb
        return hc_gbb(frame)
    elif model == 'bgsub':
        from detectors.bgsub.bgsub_detector import get_bounding_boxes as bgsub_gbb
        return bgsub_gbb(frame)
    elif model == 'ssd':
        from detectors.ssd.ssd import get_bounding_boxes as ssd_gbb
        return ssd_gbb(frame)
    elif model == 'tfoda':
        from detectors.tfoda.tfoda_detector import get_bounding_boxes as tfoda_gbb
        return tfoda_gbb(frame)
    else:
        raise Exception(
            'Invalid detector model, algorithm or API specified (options: yolo, tfoda, haarc, bgsub, ssd)'
        )

Example #3

 def get_bounding_boxes(self, frame):
     if self.model_name == 'yolo':
         from detectors.yolo.yolo_detector import get_bounding_boxes as yolo_gbb
         return yolo_gbb(frame)
     elif self.model_name == 'haarc':
         from detectors.haarc.hc_detector import get_bounding_boxes as hc_gbb
         return hc_gbb(frame)
     elif self.model_name == 'bgsub':
         from detectors.bgsub.bgsub_detector import get_bounding_boxes as bgsub_gbb
         return bgsub_gbb(frame)
     elif self.model_name == 'ssd':
         from detectors.ssd.ssd import get_bounding_boxes as ssd_gbb
         return ssd_gbb(frame)
     elif self.model_name == 'tfoda':
         from detectors.tfoda.tfoda_detector import get_bounding_boxes as tfoda_gbb
         return tfoda_gbb(frame)
     elif self.model_name == 'retinanet':
         # from detectors.retinanet.retinanet import get_bounding_boxes as retina_gbb
         # return retina_gbb(model=self.model, image=frame)
         return self.rnet.get_bounding_boxes(image=frame)
     else:
         raise Exception(
             'Invalid detector model, algorithm or API specified '
             '(options: yolo, tfoda, haarc, bgsub, ssd, retinanet)')