__init__.py 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112
  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. )
  20. from .image_classification import (
  21. ClsDatasetChecker,
  22. ClsTrainer,
  23. ClsEvaluator,
  24. ClsExportor,
  25. )
  26. from .multilabel_classification import (
  27. MLClsDatasetChecker,
  28. MLClsTrainer,
  29. MLClsEvaluator,
  30. MLClsExportor,
  31. )
  32. from .anomaly_detection import (
  33. UadDatasetChecker,
  34. UadTrainer,
  35. UadEvaluator,
  36. UadExportor,
  37. )
  38. from .general_recognition import (
  39. ShiTuRecDatasetChecker,
  40. ShiTuRecTrainer,
  41. ShiTuRecEvaluator,
  42. ShiTuRecExportor,
  43. )
  44. from .object_detection import (
  45. COCODatasetChecker,
  46. DetTrainer,
  47. DetEvaluator,
  48. DetExportor,
  49. )
  50. from .text_detection import (
  51. TextDetDatasetChecker,
  52. TextDetTrainer,
  53. TextDetEvaluator,
  54. TextDetExportor,
  55. )
  56. from .text_recognition import (
  57. TextRecDatasetChecker,
  58. TextRecTrainer,
  59. TextRecEvaluator,
  60. TextRecExportor,
  61. )
  62. from .table_recognition import (
  63. TableRecDatasetChecker,
  64. TableRecTrainer,
  65. TableRecEvaluator,
  66. TableRecExportor,
  67. )
  68. from .semantic_segmentation import (
  69. SegDatasetChecker,
  70. SegTrainer,
  71. SegEvaluator,
  72. SegExportor,
  73. )
  74. from .instance_segmentation import (
  75. COCOInstSegDatasetChecker,
  76. InstanceSegTrainer,
  77. InstanceSegEvaluator,
  78. InstanceSegExportor,
  79. )
  80. from .ts_anomaly_detection import (
  81. TSADDatasetChecker,
  82. TSADTrainer,
  83. TSADEvaluator,
  84. TSADExportor,
  85. )
  86. from .ts_classification import (
  87. TSCLSDatasetChecker,
  88. TSCLSTrainer,
  89. TSCLSEvaluator,
  90. TSCLSExportor,
  91. )
  92. from .face_recognition import (
  93. FaceRecDatasetChecker,
  94. FaceRecTrainer,
  95. FaceRecEvaluator,
  96. FaceRecExportor,
  97. )
  98. from .ts_forecast import TSFCDatasetChecker, TSFCTrainer, TSFCEvaluator
  99. from .video_classification import (
  100. VideoClsDatasetChecker,
  101. VideoClsTrainer,
  102. VideoClsEvaluator,
  103. VideoClsExportor,
  104. )