mirror of
https://github.com/qurator-spk/eynollah.git
synced 2025-06-15 23:29:54 +02:00
allow more empty imports for optional dependencies
This commit is contained in:
parent
4be89910a2
commit
46618f4229
1 changed files with 21 additions and 12 deletions
|
@ -7,6 +7,7 @@ document layout analysis (segmentation) with output in PAGE-XML
|
||||||
"""
|
"""
|
||||||
|
|
||||||
from logging import Logger
|
from logging import Logger
|
||||||
|
from difflib import SequenceMatcher as sq
|
||||||
import math
|
import math
|
||||||
import os
|
import os
|
||||||
import sys
|
import sys
|
||||||
|
@ -17,23 +18,34 @@ import warnings
|
||||||
from functools import partial
|
from functools import partial
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from multiprocessing import cpu_count
|
from multiprocessing import cpu_count
|
||||||
from loky import ProcessPoolExecutor
|
|
||||||
import gc
|
import gc
|
||||||
|
import copy
|
||||||
|
import json
|
||||||
|
|
||||||
|
from loky import ProcessPoolExecutor
|
||||||
from PIL.Image import Image
|
from PIL.Image import Image
|
||||||
from ocrd import OcrdPage
|
import xml.etree.ElementTree as ET
|
||||||
from ocrd_utils import getLogger
|
|
||||||
import cv2
|
import cv2
|
||||||
import numpy as np
|
import numpy as np
|
||||||
|
from scipy.signal import find_peaks
|
||||||
|
from scipy.ndimage import gaussian_filter1d
|
||||||
|
from numba import cuda
|
||||||
|
|
||||||
|
from ocrd import OcrdPage
|
||||||
|
from ocrd_utils import getLogger
|
||||||
|
|
||||||
|
try:
|
||||||
import torch
|
import torch
|
||||||
from difflib import SequenceMatcher as sq
|
except ImportError:
|
||||||
|
torch = None
|
||||||
|
try:
|
||||||
|
import matplotlib.pyplot as plt
|
||||||
|
except ImportError:
|
||||||
|
plt = None
|
||||||
try:
|
try:
|
||||||
from transformers import TrOCRProcessor, VisionEncoderDecoderModel
|
from transformers import TrOCRProcessor, VisionEncoderDecoderModel
|
||||||
except ImportError:
|
except ImportError:
|
||||||
TrOCRProcessor = VisionEncoderDecoderModel = None
|
TrOCRProcessor = VisionEncoderDecoderModel = None
|
||||||
from numba import cuda
|
|
||||||
import copy
|
|
||||||
from scipy.signal import find_peaks
|
|
||||||
from scipy.ndimage import gaussian_filter1d
|
|
||||||
|
|
||||||
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
|
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
|
||||||
#os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
|
#os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
|
||||||
|
@ -45,12 +57,9 @@ from tensorflow.keras.models import load_model
|
||||||
sys.stderr = stderr
|
sys.stderr = stderr
|
||||||
tf.get_logger().setLevel("ERROR")
|
tf.get_logger().setLevel("ERROR")
|
||||||
warnings.filterwarnings("ignore")
|
warnings.filterwarnings("ignore")
|
||||||
import matplotlib.pyplot as plt
|
|
||||||
# use tf1 compatibility for keras backend
|
# use tf1 compatibility for keras backend
|
||||||
from tensorflow.compat.v1.keras.backend import set_session
|
from tensorflow.compat.v1.keras.backend import set_session
|
||||||
from tensorflow.keras import layers
|
from tensorflow.keras import layers
|
||||||
import json
|
|
||||||
import xml.etree.ElementTree as ET
|
|
||||||
from tensorflow.keras.layers import StringLookup
|
from tensorflow.keras.layers import StringLookup
|
||||||
|
|
||||||
from .utils.contour import (
|
from .utils.contour import (
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue