Python create_batch_generator Examples

Example #1

File: frontend.py Project: kkmonster/K210_yolo2

 def _get_batch_generator(self, annotations, batch_size, repeat_times=1, jitter=True):
     """
     # Args
         annotations : Annotations instance
         batch_size : int
         jitter : bool
     
     # Returns
         batch_generator : BatchGenerator instance
     """
     batch_generator = create_batch_generator(annotations,
                                              self._input_size,
                                              self._yolo_network.get_grid_size(),
                                              batch_size,
                                              self._yolo_loss.anchors,
                                              repeat_times,
                                              jitter=jitter,
                                              norm=self._yolo_network.get_normalize_func())
     return batch_generator

Example #2

File: frontend.py Project: wyw636/Yolo-digit-detector

 def _get_batch_generator(self, annotations, batch_size, repeat_times=1, jitter=True):
     """
     # Args
         annotations : Annotations instance
         batch_size : int
         jitter : bool
     
     # Returns
         batch_generator : BatchGenerator instance
     """
     batch_generator = create_batch_generator(annotations,
                                              self._input_size,
                                              self._yolo_network.get_grid_size(),
                                              batch_size,
                                              self._yolo_loss.anchors,
                                              repeat_times,
                                              jitter=jitter,
                                              norm=self._yolo_network.get_normalize_func())
     return batch_generator