Skip to content
Snippets Groups Projects
Commit 6aa5bd9a authored by Solene Tarride's avatar Solene Tarride
Browse files

fix docs-build

parent d8106f34
No related branches found
No related tags found
1 merge request!76Add predicted objects to predict command
......@@ -226,7 +226,6 @@ def get_polygon(text, max_value, offset, weights, size=None, return_contours=Fal
:param text: Text piece selected with offset after splitting DAN prediction
:param max_value: Maximum "attention intensity" for parts of a text piece, used for normalization
:param offset: Offset value to get the relevant part of text piece
:param attentions: Attention weights of size (n_char, feature_height, feature_width)
:param size: Target size (width, height) to resize the coverage vector
:param return_contours: Return the contour of the current polygon (used for plotting)
"""
......@@ -285,7 +284,7 @@ def plot_attention(
:param outname: Name of the gif image
:param word_separators: List of word separators
:param line_separators: List of line separators
:param output_polygons: Whether to plot extracted polygons
:param display_polygons: Whether to plot extracted polygons
"""
height, width, _ = image.shape
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment