deepdoctection 0.31__py3-none-any.whl → 0.33__py3-none-any.whl

deepdoctection/extern/tp/tpcompat.py

@@ -18,21 +18,27 @@
 """
 Compatibility classes and methods related to Tensorpack package
 """
+from __future__ import annotations
 
+import os
 from abc import ABC, abstractmethod
-from typing import Any, List, Mapping, Tuple, Union
+from pathlib import Path
+from typing import Any, Mapping, Union
 
-from tensorpack.predict import OfflinePredictor, PredictConfig  # pylint: disable=E0401
-from tensorpack.tfutils import SmartInit  # pylint: disable=E0401
-
-# pylint: disable=import-error
-from tensorpack.train.model_desc import ModelDesc
-from tensorpack.utils.gpu import get_num_gpu
+from lazy_imports import try_import
 
 from ...utils.metacfg import AttrDict
 from ...utils.settings import ObjectTypes
+from ...utils.types import PathLikeOrStr, PixelValues
+
+with try_import() as import_guard:
+    from tensorpack.predict import OfflinePredictor, PredictConfig  # pylint: disable=E0401
+    from tensorpack.tfutils import SmartInit  # pylint: disable=E0401
+    from tensorpack.train.model_desc import ModelDesc  # pylint: disable=E0401
+    from tensorpack.utils.gpu import get_num_gpu  # pylint: disable=E0401
 
-# pylint: enable=import-error
+if not import_guard.is_successful():
+    from ...utils.mocks import ModelDesc
 
 
 class ModelDescWithConfig(ModelDesc, ABC):  # type: ignore
@@ -48,7 +54,7 @@ class ModelDescWithConfig(ModelDesc, ABC):  # type: ignore
         super().__init__()
         self.cfg = config
 
-    def get_inference_tensor_names(self) -> Tuple[List[str], List[str]]:
+    def get_inference_tensor_names(self) -> tuple[list[str], list[str]]:
         """
         Returns lists of tensor names to be used to create an inference callable. "build_graph" must create tensors
         of these names when called under inference context.
@@ -74,7 +80,7 @@ class TensorpackPredictor(ABC):
           as there is an explicit class available for this.
     """
 
-    def __init__(self, model: ModelDescWithConfig, path_weights: str, ignore_mismatch: bool) -> None:
+    def __init__(self, model: ModelDescWithConfig, path_weights: PathLikeOrStr, ignore_mismatch: bool) -> None:
         """
         :param model: Model, either as ModelDescWithConfig or derived from that class.
         :param path_weights: Model weights of the prediction config.
@@ -82,7 +88,7 @@ class TensorpackPredictor(ABC):
                                 if a pre-trained model is to be fine-tuned on a custom dataset.
         """
         self._model = model
-        self.path_weights = path_weights
+        self.path_weights = Path(path_weights)
         self.ignore_mismatch = ignore_mismatch
         self._number_gpus = get_num_gpu()
         self.predict_config = self._build_config()
@@ -95,9 +101,10 @@ class TensorpackPredictor(ABC):
         return OfflinePredictor(self.predict_config)
 
     def _build_config(self) -> PredictConfig:
+        path_weights = os.fspath(self.path_weights) if os.fspath(self.path_weights) != "." else ""
         predict_config = PredictConfig(
             model=self._model,
-            session_init=SmartInit(self.path_weights, ignore_mismatch=self.ignore_mismatch),
+            session_init=SmartInit(path_weights, ignore_mismatch=self.ignore_mismatch),
             input_names=self._model.get_inference_tensor_names()[0],
             output_names=self._model.get_inference_tensor_names()[1],
         )
@@ -107,7 +114,7 @@ class TensorpackPredictor(ABC):
     @staticmethod
     @abstractmethod
     def get_wrapped_model(
-        path_yaml: str, categories: Mapping[str, ObjectTypes], config_overwrite: Union[List[str], None]
+        path_yaml: PathLikeOrStr, categories: Mapping[int, ObjectTypes], config_overwrite: Union[list[str], None]
     ) -> ModelDescWithConfig:
         """
         Implement the config generation, its modification and instantiate a version of the model. See
@@ -116,7 +123,7 @@ class TensorpackPredictor(ABC):
         raise NotImplementedError()
 
     @abstractmethod
-    def predict(self, np_img: Any) -> Any:
+    def predict(self, np_img: PixelValues) -> Any:
         """
         Implement, how `self.tp_predictor` is invoked and raw prediction results are generated. Do use only raw
         objects and nothing, which is related to the DD API.

deepdoctection/extern/tp/tpfrcnn/__init__.py

@@ -0,0 +1,20 @@
+# -*- coding: utf-8 -*-
+# File: __init__.py
+
+# Copyright 2021 Dr. Janis Meyer. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""
+Init file for code for Tensorpack FRCNN example
+"""

deepdoctection/extern/tp/tpfrcnn/common.py

@@ -11,13 +11,17 @@ This file is modified from
 
 
 import numpy as np
-from tensorpack.dataflow.imgaug import ImageAugmentor, ResizeTransform  # pylint: disable=E0401
+from lazy_imports import try_import
 
-from ....utils.file_utils import cocotools_available
+with try_import() as import_guard:
+    from tensorpack.dataflow.imgaug import ImageAugmentor, ResizeTransform  # pylint: disable=E0401
 
-if cocotools_available():
+with try_import() as cc_import_guard:
     import pycocotools.mask as coco_mask
 
+if not import_guard.is_successful():
+    from ....utils.mocks import ImageAugmentor
+
 
 class CustomResize(ImageAugmentor):
     """

deepdoctection/extern/tp/tpfrcnn/config/__init__.py

@@ -0,0 +1,20 @@
+# -*- coding: utf-8 -*-
+# File: __init__.py
+
+# Copyright 2021 Dr. Janis Meyer. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""
+Init file for code for Tensorpack's FRCNN configs
+"""

deepdoctection/extern/tp/tpfrcnn/config/config.py

@@ -191,22 +191,25 @@ import os
 from typing import List, Mapping, Tuple
 
 import numpy as np
-from tensorpack.tfutils import collect_env_info  # pylint: disable=E0401
-from tensorpack.utils import logger  # pylint: disable=E0401
-
-# pylint: disable=import-error
-from tensorpack.utils.gpu import get_num_gpu
+from lazy_imports import try_import
 
 from .....utils.metacfg import AttrDict
-from .....utils.settings import ObjectTypes
+from .....utils.settings import TypeOrStr, get_type
+
+with try_import() as import_guard:
+    from tensorpack.tfutils import collect_env_info  # pylint: disable=E0401
+    from tensorpack.utils import logger  # pylint: disable=E0401
+
+    # pylint: disable=import-error
+    from tensorpack.utils.gpu import get_num_gpu
 
-# pylint: enable=import-error
+    # pylint: enable=import-error
 
 
 __all__ = ["train_frcnn_config", "model_frcnn_config"]
 
 
-def model_frcnn_config(config: AttrDict, categories: Mapping[str, ObjectTypes], print_summary: bool = True) -> None:
+def model_frcnn_config(config: AttrDict, categories: Mapping[int, TypeOrStr], print_summary: bool = True) -> None:
     """
     Sanity checks for Tensorpack Faster-RCNN config settings, where the focus lies on the model for predicting.
     It will update the config instance.
@@ -218,8 +221,8 @@ def model_frcnn_config(config: AttrDict, categories: Mapping[str, ObjectTypes],
 
     config.freeze(False)
 
-    categories = {str(key): categories[val] for key, val in enumerate(categories, 1)}
-    categories[0] = "BG"
+    categories = {key: get_type(categories[val]) for key, val in enumerate(categories, 1)}
+    categories[0] = get_type("background")
     config.DATA.CLASS_NAMES = list(categories.values())
     config.DATA.CLASS_DICT = categories
     config.DATA.NUM_CATEGORY = len(config.DATA.CLASS_NAMES) - 1

deepdoctection/extern/tp/tpfrcnn/modeling/__init__.py

@@ -0,0 +1,20 @@
+# -*- coding: utf-8 -*-
+# File: __init__.py
+
+# Copyright 2021 Dr. Janis Meyer. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""
+Init file for code for Tensorpack's FRCNN configs
+"""

deepdoctection/extern/tp/tpfrcnn/modeling/backbone.py

@@ -12,22 +12,30 @@ This file is modified from
 from contextlib import ExitStack, contextmanager
 
 import numpy as np
+from lazy_imports import try_import
 
 # pylint: disable=import-error
-import tensorflow as tf
-from tensorpack import tfv1
-from tensorpack.models import BatchNorm, Conv2D, MaxPooling, layer_register
-from tensorpack.tfutils import argscope
-from tensorpack.tfutils.varreplace import custom_getter_scope, freeze_variables
+
+with try_import() as import_guard:
+    import tensorflow as tf
+    from tensorpack import tfv1
+    from tensorpack.models import BatchNorm, Conv2D, MaxPooling, layer_register
+    from tensorpack.tfutils import argscope
+    from tensorpack.tfutils.varreplace import custom_getter_scope, freeze_variables
 
 # pylint: enable=import-error
 
+if not import_guard.is_successful():
+    from .....utils.mocks import layer_register
+
 
 @layer_register(log_shape=True)
-def GroupNorm(x, group=32, gamma_initializer=tf.constant_initializer(1.0)):
+def GroupNorm(x, group=32, gamma_initializer=None):
     """
     More code that reproduces the paper can be found at <https://github.com/ppwwyyxx/GroupNorm-reproduce/>.
     """
+    if gamma_initializer is None:
+        gamma_initializer = tf.constant_initializer(1.0)
     shape = x.get_shape().as_list()
     ndims = len(shape)
     assert ndims == 4, shape
@@ -63,7 +71,7 @@ def freeze_affine_getter(getter, *args, **kwargs):
     if name.endswith("/gamma") or name.endswith("/beta"):
         kwargs["trainable"] = False
         ret = getter(*args, **kwargs)
-        tf.add_to_collection(tf.GraphKeys.MODEL_VARIABLES, ret)
+        tf.add_to_collection(tf.GraphKeys.MODEL_VARIABLES, ret)  # pylint: disable=E1101
     else:
         ret = getter(*args, **kwargs)
     return ret
@@ -153,7 +161,7 @@ def get_norm(cfg, zero_init=False):
     return lambda x: norm(layer_name, x, gamma_initializer=tf.zeros_initializer() if zero_init else None)
 
 
-def resnet_shortcut(l, n_out, stride, activation=tf.identity):
+def resnet_shortcut(l, n_out, stride, activation=None):
     """
     Defining the skip connection in bottleneck
 
@@ -163,6 +171,8 @@ def resnet_shortcut(l, n_out, stride, activation=tf.identity):
     :param activation: An activation function
     :return: tf.Tensor
     """
+    if activation is None:
+        activation = tf.identity
     n_in = l.shape[1]
     if n_in != n_out:  # change dimension when channel is not the same
         return Conv2D("convshortcut", l, n_out, 1, strides=stride, activation=activation)  # pylint: disable=E1124

deepdoctection/extern/tp/tpfrcnn/modeling/generalized_rcnn.py

@@ -9,12 +9,8 @@ This file is modified from
 <https://github.com/tensorpack/tensorpack/blob/master/examples/FasterRCNN/modeling/generalized_rcnn.py>
 """
 
-# pylint: disable=import-error
-import tensorflow as tf
-from tensorpack import tfv1
-from tensorpack.models import l2_regularizer, regularize_cost
-from tensorpack.tfutils import optimizer
-from tensorpack.tfutils.summary import add_moving_summary
+
+from lazy_imports import try_import
 
 from ...tpcompat import ModelDescWithConfig
 from ..utils.box_ops import area as tf_area
@@ -40,6 +36,16 @@ from .model_frcnn import (
 from .model_mrcnn import maskrcnn_loss, unpackbits_masks
 from .model_rpn import rpn_head
 
+with try_import() as import_guard:
+    # pylint: disable=import-error
+    import tensorflow as tf
+    from tensorpack import tfv1
+    from tensorpack.models import l2_regularizer, regularize_cost
+    from tensorpack.tfutils import optimizer
+    from tensorpack.tfutils.summary import add_moving_summary
+
+    # pylint: enable=import-error
+
 
 class GeneralizedRCNN(ModelDescWithConfig):
     """

deepdoctection/extern/tp/tpfrcnn/modeling/model_box.py

@@ -11,12 +11,17 @@ This file is modified from
 from collections import namedtuple
 
 import numpy as np
+from lazy_imports import try_import
 
-# pylint: disable=import-error
-import tensorflow as tf
-from tensorpack.tfutils.scope_utils import under_name_scope
+with try_import() as import_guard:
+    # pylint: disable=import-error
+    import tensorflow as tf
+    from tensorpack.tfutils.scope_utils import under_name_scope
 
-# pylint: enable=import-error
+    # pylint: enable=import-error
+
+if not import_guard.is_successful():
+    from .....utils.mocks import under_name_scope
 
 
 @under_name_scope()
@@ -61,7 +66,7 @@ def decode_bbox_target(box_predictions, anchors, preproc_max_size):
     xbyb = box_pred_txty * waha + xaya
     x1y1 = xbyb - wbhb * 0.5
     x2y2 = xbyb + wbhb * 0.5  # (...)x1x2
-    out = tf.concat([x1y1, x2y2], axis=-2)
+    out = tf.concat([x1y1, x2y2], axis=-2)  # pylint: disable=E1123
     return tf.reshape(out, orig_shape)
 
 
@@ -88,7 +93,7 @@ def encode_bbox_target(boxes, anchors):
     # Note that here not all boxes are valid. Some may be zero
     txty = (xbyb - xaya) / waha
     twth = tf.math.log(wbhb / waha)  # may contain -inf for invalid boxes
-    encoded = tf.concat([txty, twth], axis=1)  # (-1x2x2)
+    encoded = tf.concat([txty, twth], axis=1)  # (-1x2x2)  # pylint: disable=E1123
     return tf.reshape(encoded, tf.shape(boxes))
 
 
@@ -148,7 +153,7 @@ def crop_and_resize(image, boxes, box_ind, crop_size, pad_border=True):
         n_w = spacing_w * tf.cast(crop_shape[1] - 1, tf.float32) / imshape[1]
         n_h = spacing_h * tf.cast(crop_shape[0] - 1, tf.float32) / imshape[0]
 
-        return tf.concat([ny0, nx0, ny0 + n_h, nx0 + n_w], axis=1)
+        return tf.concat([ny0, nx0, ny0 + n_h, nx0 + n_w], axis=1)  # pylint: disable=E1123
 
     image_shape = tf.shape(image)[2:]
 
@@ -208,8 +213,8 @@ class RPNAnchors(namedtuple("_RPNAnchors", ["boxes", "gt_labels", "gt_boxes"])):
         Slice anchors to the spatial size of this feature map.
         """
         shape2d = tf.shape(featuremap)[2:]  # h,w
-        slice3d = tf.concat([shape2d, [-1]], axis=0)
-        slice4d = tf.concat([shape2d, [-1, -1]], axis=0)
+        slice3d = tf.concat([shape2d, [-1]], axis=0)  # pylint: disable=E1123
+        slice4d = tf.concat([shape2d, [-1, -1]], axis=0)  # pylint: disable=E1123
         boxes = tf.slice(self.boxes, [0, 0, 0, 0], slice4d)
         gt_labels = tf.slice(self.gt_labels, [0, 0, 0], slice3d)
         gt_boxes = tf.slice(self.gt_boxes, [0, 0, 0, 0], slice4d)

deepdoctection/extern/tp/tpfrcnn/modeling/model_cascade.py

@@ -9,17 +9,20 @@ This file is modified from
 <https://github.com/tensorpack/tensorpack/blob/master/examples/FasterRCNN/modeling/model_cascade.py>
 """
 
-# pylint: disable=import-error
-import tensorflow as tf
-from tensorpack import tfv1
-from tensorpack.tfutils import get_current_tower_context
+from lazy_imports import try_import
 
 from ..utils.box_ops import area as tf_area
 from ..utils.box_ops import pairwise_iou
 from .model_box import clip_boxes
 from .model_frcnn import BoxProposals, FastRCNNHead, fastrcnn_outputs
 
-# pylint: enable=import-error
+with try_import() as import_guard:
+    # pylint: disable=import-error
+    import tensorflow as tf
+    from tensorpack import tfv1
+    from tensorpack.tfutils import get_current_tower_context
+
+    # pylint: enable=import-error
 
 
 class CascadeRCNNHead:

deepdoctection/extern/tp/tpfrcnn/modeling/model_fpn.py

@@ -12,23 +12,28 @@ This file is modified from
 import itertools
 
 import numpy as np
-
-# pylint: disable=import-error
-import tensorflow as tf
-from tensorpack import tfv1
-from tensorpack.models import Conv2D, FixedUnPooling, MaxPooling, layer_register
-from tensorpack.tfutils.argscope import argscope
-from tensorpack.tfutils.scope_utils import under_name_scope
-from tensorpack.tfutils.summary import add_moving_summary
-from tensorpack.tfutils.tower import get_current_tower_context
-from tensorpack.utils.argtools import memoized
+from lazy_imports import try_import
 
 from ..utils.box_ops import area as tf_area
 from .backbone import GroupNorm
 from .model_box import roi_align
 from .model_rpn import generate_rpn_proposals, get_all_anchors, rpn_losses
 
-# pylint: enable=import-error
+with try_import() as import_guard:
+    # pylint: disable=import-error
+    import tensorflow as tf
+    from tensorpack import tfv1
+    from tensorpack.models import Conv2D, FixedUnPooling, MaxPooling, layer_register
+    from tensorpack.tfutils.argscope import argscope
+    from tensorpack.tfutils.scope_utils import under_name_scope
+    from tensorpack.tfutils.summary import add_moving_summary
+    from tensorpack.tfutils.tower import get_current_tower_context
+    from tensorpack.utils.argtools import memoized
+
+    # pylint: enable=import-error
+
+if not import_guard.is_successful():
+    from .....utils.mocks import layer_register, memoized, under_name_scope
 
 
 @layer_register(log_shape=True)
@@ -146,9 +151,9 @@ def multilevel_roi_align(features, rcnn_boxes, resolution, fpn_anchor_strides):
             all_rois.append(roi_align(featuremap, boxes_on_featuremap, resolution))
 
     # this can fail if using TF<=1.8 with MKL build
-    all_rois = tf.concat(all_rois, axis=0)  # NCHW
+    all_rois = tf.concat(all_rois, axis=0)  # NCHW    # pylint: disable=E1123
     # Unshuffle to the original order, to match the original samples
-    level_id_perm = tf.concat(level_ids, axis=0)  # A permutation of 1~N
+    level_id_perm = tf.concat(level_ids, axis=0)  # A permutation of 1~N   # pylint: disable=E1123
     level_id_invert_perm = tf.math.invert_permutation(level_id_perm)
     all_rois = tf.gather(all_rois, level_id_invert_perm, name="output")
     return all_rois
@@ -253,8 +258,8 @@ def generate_fpn_proposals(
                 all_boxes.append(proposal_boxes)
                 all_scores.append(proposal_scores)
 
-        proposal_boxes = tf.concat(all_boxes, axis=0)  # nx4
-        proposal_scores = tf.concat(all_scores, axis=0)  # n
+        proposal_boxes = tf.concat(all_boxes, axis=0)  # nx4  # pylint: disable=E1123
+        proposal_scores = tf.concat(all_scores, axis=0)  # n  # pylint: disable=E1123
         # Here we are different from Detectron.
         # Detectron picks top-k within the batch, rather than within an image, however we do not have a batch.
         proposal_topk = tf.minimum(tf.size(proposal_scores), fpn_nms_top_k)
@@ -266,8 +271,8 @@ def generate_fpn_proposals(
                 pred_boxes_decoded = multilevel_pred_boxes[lvl]
                 all_boxes.append(tf.reshape(pred_boxes_decoded, [-1, 4]))
                 all_scores.append(tf.reshape(multilevel_label_logits[lvl], [-1]))
-        all_boxes = tf.concat(all_boxes, axis=0)
-        all_scores = tf.concat(all_scores, axis=0)
+        all_boxes = tf.concat(all_boxes, axis=0)  # pylint: disable=E1123
+        all_scores = tf.concat(all_scores, axis=0)  # pylint: disable=E1123
         proposal_boxes, proposal_scores = generate_rpn_proposals(
             all_boxes,
             all_scores,

deepdoctection/extern/tp/tpfrcnn/modeling/model_frcnn.py

@@ -8,21 +8,28 @@
 This file is modified from
 <https://github.com/tensorpack/tensorpack/blob/master/examples/FasterRCNN/modeling/model_frcnn.py>
 """
-# pylint: disable=import-error
-import tensorflow as tf
-from tensorpack import tfv1
-from tensorpack.models import Conv2D, FullyConnected, layer_register
-from tensorpack.tfutils.argscope import argscope
-from tensorpack.tfutils.common import get_tf_version_tuple
-from tensorpack.tfutils.scope_utils import under_name_scope
-from tensorpack.tfutils.summary import add_moving_summary
-from tensorpack.utils.argtools import memoized_method
+
+from lazy_imports import try_import
 
 from ..utils.box_ops import pairwise_iou
 from .backbone import GroupNorm
 from .model_box import decode_bbox_target, encode_bbox_target
 
-# pylint: enable=import-error
+with try_import() as import_guard:
+    # pylint: disable=import-error
+    import tensorflow as tf
+    from tensorpack import tfv1
+    from tensorpack.models import Conv2D, FullyConnected, layer_register
+    from tensorpack.tfutils.argscope import argscope
+    from tensorpack.tfutils.common import get_tf_version_tuple
+    from tensorpack.tfutils.scope_utils import under_name_scope
+    from tensorpack.tfutils.summary import add_moving_summary
+    from tensorpack.utils.argtools import memoized_method
+
+    # pylint: enable=import-error
+
+if not import_guard.is_successful():
+    from .....utils.mocks import layer_register, memoized_method, under_name_scope
 
 
 @under_name_scope()
@@ -72,8 +79,8 @@ def sample_fast_rcnn_targets(boxes, gt_boxes, gt_labels, frcnn_fg_thresh, frcnn_
     proposal_metrics(iou)
 
     # add ground truth as proposals as well
-    boxes = tf.concat([boxes, gt_boxes], axis=0)  # (n+m) x 4
-    iou = tf.concat([iou, tf.eye(tf.shape(gt_boxes)[0])], axis=0)  # (n+m) x m
+    boxes = tf.concat([boxes, gt_boxes], axis=0)  # (n+m) x 4  # pylint: disable=E1123
+    iou = tf.concat([iou, tf.eye(tf.shape(gt_boxes)[0])], axis=0)  # (n+m) x m  # pylint: disable=E1123
     # #proposal=n+m from now on
 
     def sample_fg_bg(iou):
@@ -107,10 +114,10 @@ def sample_fast_rcnn_targets(boxes, gt_boxes, gt_labels, frcnn_fg_thresh, frcnn_
     )
     fg_inds_wrt_gt = tf.gather(best_iou_ind, fg_inds)  # num_fg
 
-    all_indices = tf.concat([fg_inds, bg_inds], axis=0)  # indices w.r.t all n+m proposal boxes
+    all_indices = tf.concat([fg_inds, bg_inds], axis=0)  # indices w.r.t all n+m proposal boxes  # pylint: disable=E1123
     ret_boxes = tf.gather(boxes, all_indices)
 
-    ret_labels = tf.concat(
+    ret_labels = tf.concat(  # pylint: disable=E1123
         [tf.gather(gt_labels, fg_inds_wrt_gt), tf.zeros_like(bg_inds, dtype=tf.int64)],
         axis=0,
     )

deepdoctection/extern/tp/tpfrcnn/modeling/model_mrcnn.py

@@ -9,17 +9,23 @@ This file is modified from
 <https://github.com/tensorpack/tensorpack/blob/master/examples/FasterRCNN/modeling/model_mrcnn.py>
 """
 
-# pylint: disable=import-error
-import tensorflow as tf
-from tensorpack.models import Conv2D, Conv2DTranspose, layer_register
-from tensorpack.tfutils.argscope import argscope
-from tensorpack.tfutils.common import get_tf_version_tuple
-from tensorpack.tfutils.scope_utils import under_name_scope
-from tensorpack.tfutils.summary import add_moving_summary
+from lazy_imports import try_import
 
 from .backbone import GroupNorm
 
-# pylint: enable=import-error
+with try_import() as import_guard:
+    # pylint: disable=import-error
+    import tensorflow as tf
+    from tensorpack.models import Conv2D, Conv2DTranspose, layer_register
+    from tensorpack.tfutils.argscope import argscope
+    from tensorpack.tfutils.common import get_tf_version_tuple
+    from tensorpack.tfutils.scope_utils import under_name_scope
+    from tensorpack.tfutils.summary import add_moving_summary
+
+    # pylint: enable=import-error
+
+if not import_guard.is_successful():
+    from .....utils.mocks import layer_register, under_name_scope
 
 
 @under_name_scope()
@@ -41,7 +47,7 @@ def maskrcnn_loss(mask_logits, fg_labels, fg_target_masks):
 
     # add some training visualizations to tensorboard
     with tf.name_scope("mask_viz"):
-        viz = tf.concat([fg_target_masks, mask_probs], axis=1)
+        viz = tf.concat([fg_target_masks, mask_probs], axis=1)  # pylint: disable=E1123
         viz = tf.expand_dims(viz, 3)
         viz = tf.cast(viz * 255, tf.uint8, name="viz")
         tf.summary.image("mask_truth|pred", viz, max_outputs=10)
@@ -77,7 +83,7 @@ def maskrcnn_upXconv_head(feature, num_category, num_convs, norm=None, **kwargs)
     with argscope(
         [Conv2D, Conv2DTranspose],
         data_format="channels_first",
-        kernel_initializer=tf.variance_scaling_initializer(
+        kernel_initializer=tf.variance_scaling_initializer(  # pylint: disable=E1101
             scale=2.0,
             mode="fan_out",
             distribution="untruncated_normal" if get_tf_version_tuple() >= (1, 12) else "normal",
@@ -121,5 +127,7 @@ def unpackbits_masks(masks):
     assert masks.dtype == tf.uint8, masks
     bits = tf.constant((128, 64, 32, 16, 8, 4, 2, 1), dtype=tf.uint8)
     unpacked = tf.bitwise.bitwise_and(tf.expand_dims(masks, -1), bits) > 0
-    unpacked = tf.reshape(unpacked, tf.concat([tf.shape(masks)[:-1], [8 * tf.shape(masks)[-1]]], axis=0))
+    unpacked = tf.reshape(
+        unpacked, tf.concat([tf.shape(masks)[:-1], [8 * tf.shape(masks)[-1]]], axis=0)  # pylint: disable=E1123
+    )  # pylint: disable=E1123
     return unpacked

deepdoctection/extern/tp/tpfrcnn/modeling/model_rpn.py

@@ -10,19 +10,24 @@ This file is modified from
 """
 
 import numpy as np
-
-# pylint: disable=import-error
-import tensorflow as tf
-from tensorpack import tfv1
-from tensorpack.models import Conv2D, layer_register
-from tensorpack.tfutils.argscope import argscope
-from tensorpack.tfutils.scope_utils import auto_reuse_variable_scope, under_name_scope
-from tensorpack.tfutils.summary import add_moving_summary
-from tensorpack.utils.argtools import memoized
+from lazy_imports import try_import
 
 from .model_box import clip_boxes
 
-# pylint: enable=import-error
+with try_import() as import_guard:
+    # pylint: disable=import-error
+    import tensorflow as tf
+    from tensorpack import tfv1
+    from tensorpack.models import Conv2D, layer_register
+    from tensorpack.tfutils.argscope import argscope
+    from tensorpack.tfutils.scope_utils import auto_reuse_variable_scope, under_name_scope
+    from tensorpack.tfutils.summary import add_moving_summary
+    from tensorpack.utils.argtools import memoized
+
+    # pylint: enable=import-error
+
+if not import_guard.is_successful():
+    from .....utils.mocks import auto_reuse_variable_scope, layer_register, memoized, under_name_scope
 
 
 @layer_register(log_shape=True)

deepdoctection/extern/tp/tpfrcnn/predict.py

@@ -8,21 +8,26 @@
 This file is modified from
 <https://github.com/tensorpack/tensorpack/blob/master/examples/FasterRCNN/predict.py>
 """
+from __future__ import annotations
 
 from typing import List
 
-import cv2
 import numpy as np
-from tensorpack.predict.base import OfflinePredictor  # pylint: disable=E0401
+from lazy_imports import try_import
 
-from ....utils.file_utils import scipy_available
 from ....utils.transform import InferenceResize
 from ...base import DetectionResult
 from .common import clip_boxes
 
-if scipy_available():
+with try_import() as import_guard:
+    from tensorpack.predict.base import OfflinePredictor  # pylint: disable=E0401
+
+with try_import() as sp_import_guard:
     from scipy import interpolate
 
+with try_import() as cv2_import_guard:
+    import cv2
+
 
 def _scale_box(box, scale):
     w_half = (box[2] - box[0]) * 0.5