Spaces:
No application file
No application file
| """使用easydict配置适用于工程的全局参数字典 | |
| 后面可以再拆分 线上、线下、各自业务的全局字典，在使用时指定 | |
| """ | |
| from easydict import EasyDict as edct | |
| from ...utils.color_util import PolledColor | |
| from .model_cfg import ModelCfg | |
| CFG = edct() | |
| CFG.application = None | |
| CFG.update(ModelCfg) | |
| CFG.device = "cuda" | |
| CFG.model_name = "StableDiffusion" # 从model_cfg中找到对应模型的路径 | |
| CFG.pipe_name = "StableDiffusionPipeline" # 与diffusers中的pipeline类同名字符串 | |
| CFG.scheduler = None # 与diffusers中的Scheduler类同名字符串 | |
| CFG.requires_safety_checker = True # 是否使用 safety_checker | |
| CFG.template_name = "default" # 用于将输入的参数字典转化成模型prompt输入 | |
| CFG.prompt_type = "default" # prompt_type不影响代码运行，纯粹方便对prompt分类理解 | |
| CFG.target_width = 512 | |
| CFG.target_height = 512 | |
| CFG.num_inference_steps = 50 | |
| CFG.guidance_scale = 7.0 | |
| CFG.strength = 0.8 | |
| CFG.image_height_in_table = 240 | |
| CFG.try_num_util_succeed = 20 # 每个prompt任务图像生成尝试次数，不成功就放弃 | |
| CFG.seed = None # predictor 预测时的随机数生成器种子 | |
| # 一些专门定义的分隔符 | |
| # 属性字符串中有该字符串时，会通过外积运算裂变成多个任务，其他属性一模一样 | |
| CFG.task_fission_sep = "|" | |
| # 属性字符串中有该字符串时，会裂变成多个任务描述 | |
| # 如`eyes`属性中有`small,black`，则真实文本会是`small eyes, black eyes`，任务数不会发生变化 | |
| # 该部分还没有真正起作用，需要后面看怎么真正参数化，目前如果需要该功能，请在表格中记住使用`,`作为分隔符。 | |
| CFG.atrtribute_fission_sep = "," | |
| CFG.time_str_format = "%Y-%m-%d %H:%M:%S" | |
| # 输出文本图像存储方式 | |
| # 是否给存的结果增加审核列，便于候选审核 | |
| CFG.add_checker_columns = False | |
| CFG.validates = None | |
| # video_map相关算法 | |
| CFG.SemanticsAlgo = "wenlan" | |
| CFG.ClipTransitionAlgo = "TransnetV2" | |
| CFG.SscneTransitionAlgo = "SceneSeg" | |
| CFG.FaceAlgo = "insightface" | |
| CFG.TrackAlgo = "deepsort" | |
| # 剪辑相关配置 | |
| # 颜色可参考：http://www1.ynao.ac.cn/~jinhuahe/know_base/othertopics/computerissues/RGB_colortable.htm | |
| # PRODUCTION, DEVELOP, DEBUG | |
| CFG.RunningMode = "DEVELOP" | |
| CFG.TalkingDetectionTh = -0.2 | |
| CFG.FocalHumanMaxid = 3 | |
| CFG.MoviepyVideoClipReadOffset = 0.2 | |
| CFG.RecallNum = 1 | |
| CFG.RankNum = 1 | |
| # MusicInfo | |
| CFG.MSSLyricInterInterval = 4 | |
| # VideoInfo | |
| CFG.VideoStart = 0.15 | |
| CFG.VideoEnd = 0.8 | |
| # VisualizationParameters | |
| CFG.Font = "Courier" | |
| # CFG.Font = "STXinwei" # 相对秀气的字体，比较适合MV | |
| CFG.LyricFontSize = 25 | |
| # LyricTitleFontSize = CFG.LyricFontSize * 1.1 | |
| # Debug | |
| # 是否可视化clip中的帧信息，目前打开了会非常卡，默认关闭 | |
| CFG.VisFrame = False | |
| # 音乐转场点边界块，可视化颜色、宽度 | |
| CFG.MusicTransitionEdgeColor = (51, 161, 201) # 孔雀蓝 | |
| CFG.MusicTransitionEdgeSize = (20, 60) | |
| CFG.DebugFontSize = 30 | |
| CFG.DebugTextStrokeWidth = 2 | |
| CFG.DebugTextColors = ["red", "orange"] | |
| CFG.DebugTextColorsCls = PolledColor(CFG.DebugTextColors) | |
| CFG.DebugFrameTextDuration = 0.5 | |
| CFG.DebugFrameTextColor = "red" | |