__init__.py 2.6 KB

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