deepof.utils.retrieve_corners_from_image

deepof.utils.retrieve_corners_from_image(frame: ndarray, arena_type: str, cur_vid: int, videos: list)

Open a window and waits for the user to click on all corners of the polygonal arena.

The user should click on the corners in sequential order.

Parameters:
  • frame (np.ndarray) – Frame to display.

  • arena_type (str) – Type of arena to be used. Must be one of the following: “circular-manual”, “polygon-manual”.

  • cur_vid (int) – Index of the current video in the list of videos.

  • videos (list) – List of videos to be processed.

Returns:

nx2 array containing the x-y coordinates of all n corners.

Return type:

corners (np.ndarray)