mmcm/vision/__init__.py

from .human import (
    InsightfacePredictor,
    FaceTrackerByYolo5DeepSort,
    FaceClusterByInfomap,
)

# wenlan depenon detectron2, which often failed, and would be removed
from .transition.TransNetV2.transnetv2_predictor import TransNetV2Predictor

try:
    from .feature_extractor.wenlan.wenlan_predictor import (
        WenLanVisualPredictor,
    )
except:
    pass
from .transition.scene_transition_predictor import SceneTransitionPredictor
from .feature_extractor.taiyi_prefictor import TaiyiVisionFeatureExtractor
from .feature_extractor.vae_extractor import VAEFeatureExtractor

from .vis.vis_video_map import vis_video_map

from .video_map.vision_object import Role, Roles
from .video_map.video_map import VideoMap, VideoMapSeq
from .video_map.video_clip import VideoClip, VideoClipSeq
from .video_map.video_meta_info import VideoMetaInfo
from .video_map.load_video_map import load_video_map

from .utils.path_util import (
    get_video_signature,
    get_video_path_dct,
    get_dir_file_map,
    get_video_map_path_dct,
    get_video_emd_path_dct,
)