Ma/video-subtitle-remover
1
0
Fork 0
mirror of https://github.com/YaoFANGUK/video-subtitle-remover.git synced 2026-03-07 03:07:33 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

继续修复bug

Browse Source
This commit is contained in:
YaoFANGUK
2023-12-27 20:32:00 +08:00
parent f92a483717
commit 313c3d37a7
4 changed files with 74 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
backend/tools/inpaint_tools.py
View File

@@ -78,8 +78,16 @@ def create_mask(size, coords_list):
for coords in coords_list:
xmin, xmax, ymin, ymax = coords
# 为了避免框过小，放大10个像素
cv2.rectangle(mask, (xmin - config.SUBTITLE_AREA_DEVIATION_PIXEL, ymin - config.SUBTITLE_AREA_DEVIATION_PIXEL),
(xmax + config.SUBTITLE_AREA_DEVIATION_PIXEL, ymax + config.SUBTITLE_AREA_DEVIATION_PIXEL), (255, 255, 255), thickness=-1)
x1 = xmin - config.SUBTITLE_AREA_DEVIATION_PIXEL
if x1 < 0:
x1 = 0
y1 = ymin - config.SUBTITLE_AREA_DEVIATION_PIXEL
if y1 < 0:
y1 = 0
x2 = xmax + config.SUBTITLE_AREA_DEVIATION_PIXEL
y2 = ymax + config.SUBTITLE_AREA_DEVIATION_PIXEL
cv2.rectangle(mask, (x1, y1),
(x2, y2), (255, 255, 255), thickness=-1)
return mask