修复自动检测文本时若mask高度大于宽度进程卡住的bug

This commit is contained in:
YaoFANGUK
2023-12-27 09:22:25 +08:00
parent a183178a59
commit 4d3d4b59bd
3 changed files with 15 additions and 5 deletions

View File

@@ -33,8 +33,8 @@ class STTNInpaint:
# 模型输入用的宽和高
self.model_input_width, self.model_input_height = 640, 120
# 2. 设置相连帧数
self.neighbor_stride = 5
self.ref_length = 5
self.neighbor_stride = config.NEIGHBOR_STRIDE
self.ref_length = config.REFERENCE_LENGTH
def __call__(self, input_frames: List[np.ndarray], input_mask: np.ndarray):
"""