__init__.py 2.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495
  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. TSADPredictor,
  77. )
  78. from .ts_classification import (
  79. TSCLSDatasetChecker,
  80. TSCLSTrainer,
  81. TSCLSEvaluator,
  82. TSCLSPredictor,
  83. )
  84. from .ts_forecast import TSFCDatasetChecker, TSFCTrainer, TSFCEvaluator, TSFCPredictor
  85. from .base.predictor.transforms import image_common
  86. from .image_classification import transforms as cls_transforms
  87. from .object_detection import transforms as det_transforms
  88. from .text_detection import transforms as text_det_transforms
  89. from .text_recognition import transforms as text_rec_transforms
  90. from .table_recognition import transforms as table_rec_transforms
  91. from .semantic_segmentation import transforms as seg_transforms
  92. from .instance_segmentation import transforms as instance_seg_transforms