Files
chaojie-ComfyUI_StreamingT2V/thirdparty/modelscope/trainers/cv/__init__.py
T
2024-04-08 06:24:30 +08:00

47 lines
2.0 KiB
Python

# These codes are copied from modelscope revision c58451baead80d83281f063d12fb377fad415257
# Copyright (c) Alibaba, Inc. and its affiliates.
from typing import TYPE_CHECKING
from modelscope.utils.import_utils import LazyImportModule
if TYPE_CHECKING:
from .image_instance_segmentation_trainer import \
ImageInstanceSegmentationTrainer
from .image_portrait_enhancement_trainer import ImagePortraitEnhancementTrainer
from .movie_scene_segmentation_trainer import MovieSceneSegmentationTrainer
from .image_inpainting_trainer import ImageInpaintingTrainer
from .referring_video_object_segmentation_trainer import ReferringVideoObjectSegmentationTrainer
from .image_defrcn_fewshot_detection_trainer import ImageDefrcnFewshotTrainer
from .cartoon_translation_trainer import CartoonTranslationTrainer
from .ocr_recognition_trainer import OCRRecognitionTrainer
from .nerf_recon_acc_trainer import NeRFReconAccTrainer
from .vision_efficient_tuning_trainer import VisionEfficientTuningTrainer
else:
_import_structure = {
'image_instance_segmentation_trainer':
['ImageInstanceSegmentationTrainer'],
'image_portrait_enhancement_trainer':
['ImagePortraitEnhancementTrainer'],
'movie_scene_segmentation_trainer': ['MovieSceneSegmentationTrainer'],
'image_inpainting_trainer': ['ImageInpaintingTrainer'],
'referring_video_object_segmentation_trainer':
['ReferringVideoObjectSegmentationTrainer'],
'image_defrcn_fewshot_detection_trainer':
['ImageDefrcnFewshotTrainer'],
'cartoon_translation_trainer': ['CartoonTranslationTrainer'],
'ocr_recognition_trainer': ['OCRRecognitionTrainer'],
'nerf_recon_acc_trainer': ['NeRFReconAccTrainer'],
'vision_efficient_tuning_trainer': ['VisionEfficientTuningTrainer'],
}
import sys
sys.modules[__name__] = LazyImportModule(
__name__,
globals()['__file__'],
_import_structure,
module_spec=__spec__,
extra_objects={},
)