allow more empty imports for optional dependencies

pull/148/head
Robert Sachunsky 2 weeks ago
parent 4be89910a2
commit 46618f4229

@ -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
import torch from scipy.signal import find_peaks
from difflib import SequenceMatcher as sq from scipy.ndimage import gaussian_filter1d
from numba import cuda
from ocrd import OcrdPage
from ocrd_utils import getLogger
try:
import torch
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…
Cancel
Save