From 9a65c17a506fd0d0d741a1f5d720d83556658686 Mon Sep 17 00:00:00 2001 From: YaoFANGUK Date: Fri, 5 Jan 2024 14:39:05 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E5=A4=87=E6=B3=A8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- backend/config.py | 26 ++++++++++++++++++++------ 1 file changed, 20 insertions(+), 6 deletions(-) diff --git a/backend/config.py b/backend/config.py index b8bae91..34d1ee8 100644 --- a/backend/config.py +++ b/backend/config.py @@ -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