From 156ece6ca395810915e32db6e9aa5e1322d61243 Mon Sep 17 00:00:00 2001
From: starride-teklia <starride@teklia.com>
Date: Fri, 7 Apr 2023 11:19:56 +0000
Subject: [PATCH] Convert filename to pathlib.Path
---
dan/predict/prediction.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/dan/predict/prediction.py b/dan/predict/prediction.py
index 09e7a7ba..30c9ff5f 100644
--- a/dan/predict/prediction.py
+++ b/dan/predict/prediction.py
@@ -2,6 +2,7 @@
import os
import pickle
+from pathlib import Path
import cv2
import numpy as np
@@ -100,6 +101,7 @@ class DAN:
extract_objects=False,
word_separators=["\n", " "],
line_separators=["\n"],
+ start_token=None,
):
"""
Run prediction on an input image.
@@ -352,4 +354,4 @@ def run(
json_filename = f"{output}/{image.stem}.json"
logger.info(f"Saving JSON prediction in {json_filename}")
- save_json(json_filename, result)
+ save_json(Path(json_filename), result)
--
GitLab