From 59ab6de96d9730be63576e0db3172cb96d98c9bc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?M=C3=A9lodie=20Boillet?= <boillet@teklia.com>
Date: Wed, 2 Aug 2023 14:51:00 +0200
Subject: [PATCH] Apply 2bb85c50
---
dan/manager/dataset.py | 3 ++-
dan/manager/ocr.py | 1 -
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/dan/manager/dataset.py b/dan/manager/dataset.py
index d441c6a1..92de20dd 100644
--- a/dan/manager/dataset.py
+++ b/dan/manager/dataset.py
@@ -1,4 +1,5 @@
# -*- coding: utf-8 -*-
+import copy
import json
import os
@@ -112,7 +113,7 @@ class OCRDataset(Dataset):
def get_sample_img(self, i):
"""
- Get image by index
+ Compute the final image size and position after feature extraction
"""
if self.load_in_memory:
return self.samples[i]["img"]
diff --git a/dan/manager/ocr.py b/dan/manager/ocr.py
index c0e832b3..d7f72ea3 100644
--- a/dan/manager/ocr.py
+++ b/dan/manager/ocr.py
@@ -1,5 +1,4 @@
# -*- coding: utf-8 -*-
-import copy
import os
import pickle
import random
--
GitLab