__init__.py 1.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  1. # Copyright (c) Alibaba, Inc. and its affiliates.
  2. from typing import TYPE_CHECKING
  3. from modelscope.utils.import_utils import LazyImportModule
  4. if TYPE_CHECKING:
  5. from .image_instance_segmentation_trainer import \
  6. ImageInstanceSegmentationTrainer
  7. from .image_portrait_enhancement_trainer import ImagePortraitEnhancementTrainer
  8. from .movie_scene_segmentation_trainer import MovieSceneSegmentationTrainer
  9. from .image_inpainting_trainer import ImageInpaintingTrainer
  10. from .referring_video_object_segmentation_trainer import ReferringVideoObjectSegmentationTrainer
  11. from .image_defrcn_fewshot_detection_trainer import ImageDefrcnFewshotTrainer
  12. from .cartoon_translation_trainer import CartoonTranslationTrainer
  13. from .ocr_recognition_trainer import OCRRecognitionTrainer
  14. from .nerf_recon_acc_trainer import NeRFReconAccTrainer
  15. from .vision_efficient_tuning_trainer import VisionEfficientTuningTrainer
  16. else:
  17. _import_structure = {
  18. 'image_instance_segmentation_trainer':
  19. ['ImageInstanceSegmentationTrainer'],
  20. 'image_portrait_enhancement_trainer':
  21. ['ImagePortraitEnhancementTrainer'],
  22. 'movie_scene_segmentation_trainer': ['MovieSceneSegmentationTrainer'],
  23. 'image_inpainting_trainer': ['ImageInpaintingTrainer'],
  24. 'referring_video_object_segmentation_trainer':
  25. ['ReferringVideoObjectSegmentationTrainer'],
  26. 'image_defrcn_fewshot_detection_trainer':
  27. ['ImageDefrcnFewshotTrainer'],
  28. 'cartoon_translation_trainer': ['CartoonTranslationTrainer'],
  29. 'ocr_recognition_trainer': ['OCRRecognitionTrainer'],
  30. 'nerf_recon_acc_trainer': ['NeRFReconAccTrainer'],
  31. 'vision_efficient_tuning_trainer': ['VisionEfficientTuningTrainer'],
  32. }
  33. import sys
  34. sys.modules[__name__] = LazyImportModule(
  35. __name__,
  36. globals()['__file__'],
  37. _import_structure,
  38. module_spec=__spec__,
  39. extra_objects={},
  40. )