修改备注

This commit is contained in:
YaoFANGUK
2024-01-05 14:39:05 +08:00
parent 3ce8d7409b
commit 9a65c17a50

View File

@@ -87,13 +87,27 @@ PIXEL_TOLERANCE_X = 20 # 允许检测框横向偏差的像素点数
# ×××××××××× InpaintMode.STTN算法设置 start ××××××××××
# 以下参数仅适用STTN算法时才生效
# 是否使用跳过检测,跳过字幕检测会省去很大时间,但是可能误伤无字幕的视频帧
# 注意将STTN_SKIP_DETECTION设置为True可能会导致去除的字幕漏了
STTN_SKIP_DETECTION = False
# 相邻帧数, 调大会增加显存占用,效果变好
STTN_NEIGHBOR_STRIDE = 10
# 参考帧长度, 调大会增加显存占用,效果变好
STTN_SKIP_DETECTION = True
"""
STTN_SKIP_DETECTION
含义:是否使用跳过检测
效果设置为True跳过字幕检测会省去很大时间但是可能误伤无字幕的视频帧或者会导致去除的字幕漏了
"""
# 参考帧步长
STTN_NEIGHBOR_STRIDE = 5
"""
STTN_NEIGHBOR_STRIDE
含义:相邻帧数步长, 如果我们需要为第50帧填充缺失的区域STTN_NEIGHBOR_STRIDE=5那么算法可能会使用第45帧、第40帧等作为参照。
效果:用于控制参考帧选择的密度,较大的步长意味着使用更少、更分散的参考帧,较小的步长意味着使用更多、更集中的参考帧。
"""
# 参考帧长度(数量)
STTN_REFERENCE_LENGTH = 10
"""
STTN_REFERENCE_LENGTH
含义参数帧数量STTN算法会查看每个待修复帧的前后若干帧来获得用于修复的上下文信息
效果:调大会增加显存占用,处理效果变好,但是处理速度变慢
"""
# 设置STTN算法最大同时处理的帧数量设置越大速度越慢但效果越好
# 要保证STTN_MAX_LOAD_NUM大于STTN_NEIGHBOR_STRIDE和STTN_REFERENCE_LENGTH
STTN_MAX_LOAD_NUM = 30