tools/accuracy_checker/accuracy_checker/postprocessor/__init__.py

   1 """
   2 Copyright (c) 2019 Intel Corporation
   3
   4 Licensed under the Apache License, Version 2.0 (the "License");
   5 you may not use this file except in compliance with the License.
   6 You may obtain a copy of the License at
   7
   8       http://www.apache.org/licenses/LICENSE-2.0
   9
  10 Unless required by applicable law or agreed to in writing, software
  11 distributed under the License is distributed on an "AS IS" BASIS,
  12 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 See the License for the specific language governing permissions and
  14 limitations under the License.
  15 """
  16
  17 from .postprocessing_executor import PostprocessingExecutor
  18
  19 from .filter import (
  20     FilterPostprocessor,
  21
  22     FilterByHeightRange,
  23     FilterByLabels,
  24     FilterByMinConfidence,
  25     FilterEmpty,
  26     FilterByVisibility,
  27     FilterByAspectRatio
  28 )
  29
  30 from .cast_to_int import CastToInt
  31 from .clip_boxes import ClipBoxes
  32 from .nms import NMS
  33 from .resize_prediction_boxes import ResizePredictionBoxes
  34 from .correct_yolo_v2_boxes import CorrectYoloV2Boxes
  35 from .resize_segmentation_mask import ResizeSegmentationMask
  36 from .encode_segmentation_mask import EncodeSegMask
  37 from .normalize_landmarks_points import NormalizeLandmarksPoints
  38 from .clip_points import ClipPoints
  39 from .extend_segmentation_mask import ExtendSegmentationMask
  40 from .zoom_segmentation_mask import ZoomSegMask
  41 from .crop_segmentation_mask import CropSegmentationMask
  42 from .clip_segmentation_mask import ClipSegmentationMask
  43
  44 __all__ = [
  45     'PostprocessingExecutor',
  46
  47     'FilterPostprocessor',
  48     'FilterByHeightRange',
  49     'FilterByLabels',
  50     'FilterByMinConfidence',
  51     'FilterEmpty',
  52     'FilterByVisibility',
  53     'FilterByAspectRatio',
  54
  55     'CastToInt',
  56     'ClipBoxes',
  57     'NMS',
  58     'ResizePredictionBoxes',
  59     'CorrectYoloV2Boxes',
  60
  61     'ResizeSegmentationMask',
  62     'EncodeSegMask',
  63     'ExtendSegmentationMask',
  64     'ZoomSegMask',
  65     'CropSegmentationMask',
  66     'ClipSegmentationMask',
  67
  68     'NormalizeLandmarksPoints'
  69 ]