예제 #1
0
    def build_corner_storage(self) -> CornerStorage:
        self.process_all_events()

        frame_corners = []
        for frame_id in range(self._n_frames):
            ids = self._frame_point_ids[frame_id]
            sizes = self._point_sizes[ids]

            points2d = []
            for point_id in ids:
                i = frame_id - self._point_first_frames[point_id]
                points2d += [np.expand_dims(self._points2d[point_id][i], axis=0)]

            points2d = np.concatenate(points2d, axis=0)

            frame_corners += [FrameCorners(ids, points2d, sizes)]

        return StorageImpl(frame_corners)
예제 #2
0
 def build_corner_storage(self):
     return StorageImpl(item[1] for item in sorted(self._corners.items()))