42 lines
1.5 KiB
Python
42 lines
1.5 KiB
Python
from .nodes.traverse_folder import TraverseFolder
|
|
from .nodes.unload_all_models import UnloadAllModels
|
|
from .nodes.string_empty_judgement import StringEmptyJudgement
|
|
from .nodes.compute_video_point import VideoStartPointDurationCompute
|
|
from .nodes.cos import COSUpload, COSDownload
|
|
from .nodes.face_detect import FaceDetect
|
|
from .nodes.face_extract import FaceExtract
|
|
from .nodes.log2db import LogToDB
|
|
from .nodes.videocut import VideoCut
|
|
from .nodes.vod2local import VodToLocalNode
|
|
|
|
# A dictionary that contains all nodes you want to export with their names
|
|
# NOTE: names should be globally unique
|
|
NODE_CLASS_MAPPINGS = {
|
|
"FaceOccDetect": FaceDetect,
|
|
"FaceExtract": FaceExtract,
|
|
"COSUpload": COSUpload,
|
|
"COSDownload": COSDownload,
|
|
"VideoCutCustom": VideoCut,
|
|
"VodToLocal": VodToLocalNode,
|
|
"LogToDB": LogToDB,
|
|
"VideoPointCompute": VideoStartPointDurationCompute,
|
|
"StringEmptyJudgement": StringEmptyJudgement,
|
|
"unloadAllModels": UnloadAllModels,
|
|
"TraverseFolder": TraverseFolder,
|
|
}
|
|
|
|
# A dictionary that contains the friendly/humanly readable titles for the nodes
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"FaceOccDetect": "面部遮挡检测",
|
|
"FaceExtract": "面部提取",
|
|
"COSUpload": "COS上传",
|
|
"COSDownload": "COS下载",
|
|
"VideoCutCustom": "视频剪裁",
|
|
"VodToLocal": "腾讯云VOD下载",
|
|
"LogToDB": "状态持久化DB",
|
|
"VideoPointCompute": "视频帧位计算",
|
|
"StringEmptyJudgement": "字符串是否为空",
|
|
"unloadAllModels": "卸载所有已加载模型",
|
|
"TraverseFolder": "遍历文件夹"
|
|
}
|