INNER CODE UNIT · Python
total_size
GeekAlexis/FastMOT · fastmot/detector.py:127
total_size = (tiling_grid - 1) * step_size + tile_size
total_size = np.rint(total_size).astype(int)
tiles = np.array([to_tlbr((c * step_size[0], r * step_size[1], *tile_size))
for r in range(tiling_grid[1]) for c in range(tiling_grid[0])])
return tiles, tuple(total_size)
def _merge_dets(self, detections, tile_ids):
detections = np.fromiter(detections, DET_DTYPE, len(detections)).view(np.recarray)
tile_ids = np.fromiter(tile_ids, int, len(tile_ids))
if len(detections) == 0:
return detections
detections = self._merge(detections, tile_ids, self.batch_size, self.merge_thresh)
return detections.view(np.recarray)
@staticmethod
@nb.njit(parallel=True, fastmath=True, cache=True)
def _normalize(frame, tiles, out):
imgs = multi_crop(frame, tiles)