__init__.py 2.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109
  1. # copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
  2. #
  3. # Licensed under the Apache License, Version 2.0 (the "License");
  4. # you may not use this file except in compliance with the License.
  5. # You may obtain a copy of the License at
  6. #
  7. # http://www.apache.org/licenses/LICENSE-2.0
  8. #
  9. # Unless required by applicable law or agreed to in writing, software
  10. # distributed under the License is distributed on an "AS IS" BASIS,
  11. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. # See the License for the specific language governing permissions and
  13. # limitations under the License.
  14. from .base import (
  15. build_dataset_checker,
  16. build_trainer,
  17. build_evaluater,
  18. build_exportor,
  19. build_predictor,
  20. create_model,
  21. PaddleInferenceOption,
  22. )
  23. from .image_classification import (
  24. ClsDatasetChecker,
  25. ClsTrainer,
  26. ClsEvaluator,
  27. ClsExportor,
  28. ClsPredictor,
  29. )
  30. from .general_recognition import (
  31. ShiTuRecDatasetChecker,
  32. ShiTuRecTrainer,
  33. ShiTuRecEvaluator,
  34. ShiTuRecExportor,
  35. )
  36. from .object_detection import (
  37. COCODatasetChecker,
  38. DetTrainer,
  39. DetEvaluator,
  40. DetExportor,
  41. DetPredictor,
  42. )
  43. from .text_detection import (
  44. TextDetDatasetChecker,
  45. TextDetTrainer,
  46. TextDetEvaluator,
  47. TextDetExportor,
  48. TextDetPredictor,
  49. )
  50. from .text_recognition import (
  51. TextRecDatasetChecker,
  52. TextRecTrainer,
  53. TextRecEvaluator,
  54. TextRecExportor,
  55. TextRecPredictor,
  56. )
  57. from .table_recognition import (
  58. TableRecDatasetChecker,
  59. TableRecTrainer,
  60. TableRecEvaluator,
  61. TableRecExportor,
  62. TableRecPredictor,
  63. )
  64. from .semantic_segmentation import (
  65. SegDatasetChecker,
  66. SegTrainer,
  67. SegEvaluator,
  68. SegExportor,
  69. SegPredictor,
  70. )
  71. from .instance_segmentation import (
  72. COCOInstSegDatasetChecker,
  73. InstanceSegTrainer,
  74. InstanceSegEvaluator,
  75. InstanceSegExportor,
  76. InstanceSegPredictor,
  77. )
  78. from .ts_anomaly_detection import (
  79. TSADDatasetChecker,
  80. TSADTrainer,
  81. TSADEvaluator,
  82. TSADExportor,
  83. TSADPredictor,
  84. )
  85. from .ts_classification import (
  86. TSCLSDatasetChecker,
  87. TSCLSTrainer,
  88. TSCLSEvaluator,
  89. TSCLSExportor,
  90. TSCLSPredictor,
  91. )
  92. from .ts_forecast import TSFCDatasetChecker, TSFCTrainer, TSFCEvaluator, TSFCPredictor
  93. from .image_unwarping import WarpPredictor
  94. from .base.predictor.transforms import image_common
  95. from .image_classification import transforms as cls_transforms
  96. from .object_detection import transforms as det_transforms
  97. from .text_detection import transforms as text_det_transforms
  98. from .text_recognition import transforms as text_rec_transforms
  99. from .table_recognition import transforms as table_rec_transforms
  100. from .semantic_segmentation import transforms as seg_transforms
  101. from .instance_segmentation import transforms as instance_seg_transforms
  102. from .image_unwarping import transforms as image_unwarping_transforms