__init__.py 3.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114
  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 .anomaly_detection import (
  31. UadDatasetChecker,
  32. UadTrainer,
  33. UadEvaluator,
  34. UadExportor,
  35. UadPredictor,
  36. )
  37. from .general_recognition import (
  38. ShiTuRecDatasetChecker,
  39. ShiTuRecTrainer,
  40. ShiTuRecEvaluator,
  41. ShiTuRecExportor,
  42. )
  43. from .object_detection import (
  44. COCODatasetChecker,
  45. DetTrainer,
  46. DetEvaluator,
  47. DetExportor,
  48. DetPredictor,
  49. )
  50. from .text_detection import (
  51. TextDetDatasetChecker,
  52. TextDetTrainer,
  53. TextDetEvaluator,
  54. TextDetExportor,
  55. TextDetPredictor,
  56. )
  57. from .text_recognition import (
  58. TextRecDatasetChecker,
  59. TextRecTrainer,
  60. TextRecEvaluator,
  61. TextRecExportor,
  62. TextRecPredictor,
  63. )
  64. from .table_recognition import (
  65. TableRecDatasetChecker,
  66. TableRecTrainer,
  67. TableRecEvaluator,
  68. TableRecExportor,
  69. TableRecPredictor,
  70. )
  71. from .semantic_segmentation import (
  72. SegDatasetChecker,
  73. SegTrainer,
  74. SegEvaluator,
  75. SegExportor,
  76. SegPredictor,
  77. )
  78. from .instance_segmentation import (
  79. COCOInstSegDatasetChecker,
  80. InstanceSegTrainer,
  81. InstanceSegEvaluator,
  82. InstanceSegExportor,
  83. InstanceSegPredictor,
  84. )
  85. from .ts_anomaly_detection import (
  86. TSADDatasetChecker,
  87. TSADTrainer,
  88. TSADEvaluator,
  89. TSADExportor,
  90. TSADPredictor,
  91. )
  92. from .ts_classification import (
  93. TSCLSDatasetChecker,
  94. TSCLSTrainer,
  95. TSCLSEvaluator,
  96. TSCLSExportor,
  97. TSCLSPredictor,
  98. )
  99. from .ts_forecast import TSFCDatasetChecker, TSFCTrainer, TSFCEvaluator, TSFCPredictor
  100. from .image_unwarping import WarpPredictor
  101. from .base.predictor.transforms import image_common
  102. from .image_classification import transforms as cls_transforms
  103. from .object_detection import transforms as det_transforms
  104. from .text_detection import transforms as text_det_transforms
  105. from .text_recognition import transforms as text_rec_transforms
  106. from .table_recognition import transforms as table_rec_transforms
  107. from .semantic_segmentation import transforms as seg_transforms
  108. from .instance_segmentation import transforms as instance_seg_transforms
  109. from .image_unwarping import transforms as image_unwarping_transforms