Kevin Hu commited on
Commit
0826715
·
1 Parent(s): d9dc926

remove dependency (#1536)

Browse files

### What problem does this PR solve?

#702
### Type of change
- [x] Refactoring

deepdoc/vision/ocr.py CHANGED
@@ -23,7 +23,6 @@ import numpy as np
23
  import onnxruntime as ort
24
 
25
  from .postprocess import build_post_process
26
- from rag.settings import cron_logger
27
 
28
 
29
  def transform(data, ops=None):
@@ -566,9 +565,6 @@ class OCR(object):
566
  end = time.time()
567
  time_dict['all'] = end - start
568
  return None, None, time_dict
569
- else:
570
- cron_logger.debug("dt_boxes num : {}, elapsed : {}".format(
571
- len(dt_boxes), elapse))
572
 
573
  return zip(self.sorted_boxes(dt_boxes), [
574
  ("", 0) for _ in range(len(dt_boxes))])
@@ -597,9 +593,7 @@ class OCR(object):
597
  end = time.time()
598
  time_dict['all'] = end - start
599
  return None, None, time_dict
600
- else:
601
- cron_logger.debug("dt_boxes num : {}, elapsed : {}".format(
602
- len(dt_boxes), elapse))
603
  img_crop_list = []
604
 
605
  dt_boxes = self.sorted_boxes(dt_boxes)
@@ -612,8 +606,6 @@ class OCR(object):
612
  rec_res, elapse = self.text_recognizer(img_crop_list)
613
 
614
  time_dict['rec'] = elapse
615
- cron_logger.debug("rec_res num : {}, elapsed : {}".format(
616
- len(rec_res), elapse))
617
 
618
  filter_boxes, filter_rec_res = [], []
619
  for box, rec_result in zip(dt_boxes, rec_res):
 
23
  import onnxruntime as ort
24
 
25
  from .postprocess import build_post_process
 
26
 
27
 
28
  def transform(data, ops=None):
 
565
  end = time.time()
566
  time_dict['all'] = end - start
567
  return None, None, time_dict
 
 
 
568
 
569
  return zip(self.sorted_boxes(dt_boxes), [
570
  ("", 0) for _ in range(len(dt_boxes))])
 
593
  end = time.time()
594
  time_dict['all'] = end - start
595
  return None, None, time_dict
596
+
 
 
597
  img_crop_list = []
598
 
599
  dt_boxes = self.sorted_boxes(dt_boxes)
 
606
  rec_res, elapse = self.text_recognizer(img_crop_list)
607
 
608
  time_dict['rec'] = elapse
 
 
609
 
610
  filter_boxes, filter_rec_res = [], []
611
  for box, rec_result in zip(dt_boxes, rec_res):
deepdoc/vision/recognizer.py CHANGED
@@ -19,7 +19,6 @@ from huggingface_hub import snapshot_download
19
 
20
  from api.utils.file_utils import get_project_base_directory
21
  from .operators import *
22
- from rag.settings import cron_logger
23
 
24
 
25
  class Recognizer(object):
@@ -340,7 +339,6 @@ class Recognizer(object):
340
  if score < thr:
341
  continue
342
  if clsid >= len(self.label_list):
343
- cron_logger.warning(f"bad category id")
344
  continue
345
  bb.append({
346
  "type": self.label_list[clsid].lower(),
 
19
 
20
  from api.utils.file_utils import get_project_base_directory
21
  from .operators import *
 
22
 
23
 
24
  class Recognizer(object):
 
339
  if score < thr:
340
  continue
341
  if clsid >= len(self.label_list):
 
342
  continue
343
  bb.append({
344
  "type": self.label_list[clsid].lower(),