From 6aa5bd9a13c1260716fc9eef49857f9e92632b99 Mon Sep 17 00:00:00 2001
From: starride-teklia <starride@teklia.com>
Date: Tue, 11 Apr 2023 13:13:12 +0000
Subject: [PATCH] fix docs-build
---
dan/predict/attention.py | 3 +--
1 file changed, 1 insertion(+), 2 deletions(-)
diff --git a/dan/predict/attention.py b/dan/predict/attention.py
index 117adb78..4ef781c3 100644
--- a/dan/predict/attention.py
+++ b/dan/predict/attention.py
@@ -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
--
GitLab