Spaces:
No application file
No application file
File size: 3,159 Bytes
6755a2d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 |
"""使用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"
|