dan/predict/prediction.py

@@ -92,6 +92,7 @@ class DAN:
         input_sizes,
         confidences=False,
         attentions=False,
+        start_token=None,
     ):
         """
         Run prediction on an input image.
@@ -102,7 +103,9 @@ class DAN:
         """
         input_tensor = input_tensor.to(self.device)
 
-        start_token = len(self.charset) + 1
+        start_token = (
+            self.charset.index(start_token) if start_token else len(self.charset) + 1
+        )
         end_token = len(self.charset)
 
         # Run the prediction.

doc-requirements.txt

@@ -3,5 +3,5 @@ doc8==1.1.1
 mkdocs==1.4.2
 mkdocs-material==9.1.3
 mkdocstrings==0.20.0
-mkdocstrings-python==0.8.3
+mkdocstrings-python==0.9.0
 recommonmark==0.7.1

requirements.txt

 arkindex-export==0.1.2
-boto3==1.26.98
+boto3==1.26.106
 editdistance==0.6.2
 fontTools==4.39.2
 imageio==2.26.1
 mlflow==2.2.2
-networkx==3.0
+networkx==3.1
 numpy==1.23.5
 opencv-python==4.7.0.72
 PyYAML==6.0
 scipy==1.10.1
-tensorboard==2.12.0
+tensorboard==2.12.1
 torch==2.0.0
 torchvision==0.15.1
 tqdm==4.65.0