Skip to content

Commit

Permalink
Update narrator.py
Browse files Browse the repository at this point in the history
  • Loading branch information
Reza Mansouri authored Sep 12, 2022
1 parent ecc49a6 commit 7a1086c
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions src/tell_a_vision/narrator.py
Original file line number Diff line number Diff line change
Expand Up @@ -58,8 +58,7 @@ def get_narration(classes, class_labels, ranks, locations, rank_to_distance_labe
"""
:param classes: array of shape (n, ) with each element corresponding to the index of the object's class in class_labels
:param class_labels: list of class labels of objects in your dataset. For example: ['car', 'bike', 'person', 'truck'].
ranks: output of tv.Ruler.get_rank(). Array of shape (n, ) with each element being 0, 1, 2, or 3 representing an object's size/distance
:param ranks: output of tv.locate(). Array of shape (n, 2) containing the locations of the boxes: [h_location, v_location]
:param ranks: output of tv.Ruler.get_rank(). Array of shape (n, ) with each element being 0, 1, 2, or 3 representing an object's size/distance
:param locations: output of tv.locate(). Array of shape (n, 2) containing the locations of the boxes: [h_location, v_location]
:param rank_to_distance_labels: tuple of length 4 describing objects' distance/size according to their ranks (quartile intervals)
:param h_location_to_lr_labels: tuple of length 3 describing objects' horizontal location
Expand Down

0 comments on commit 7a1086c

Please sign in to comment.