新增视频inpaint方法

This commit is contained in:
YaoFANGUK
2023-12-26 10:10:43 +08:00
parent 9aeb8f8860
commit 41a95dac0f
3 changed files with 107 additions and 33 deletions

View File

@@ -35,10 +35,12 @@ THRESHOLD_HEIGHT_DIFFERENCE = 20
MAX_PROCESS_NUM = 70
# 【根据自己内存大小设置应该大于等于MAX_PROCESS_NUM】
MAX_LOAD_NUM = 200
# 是否开启精细模式开启精细模式将消耗大量GPU显存如果您的显卡显存较少建议设置为False
ACCURATE_MODE = True
# 是否开启快速模型不保证inpaint效果
FAST_MODE = False
# 模式列表请根据自己需求选择inpiant模式
# ACCURATE模式将消耗大量GPU显存如果您的显卡显存较少建议设置为NORMAL
MODE_LIST = ['FAST', 'NORMAL', 'ACCURATE']
MODE = 'NORMAL'
# 如果仅需要去除文字区域则使用FAST
SUPER_FAST = False
# ×××××××××××××××××××× [可以改] start ××××××××××××××××××××
@@ -73,8 +75,6 @@ if 'ffmpeg.exe' not in os.listdir(os.path.join(BASE_DIR, '', 'ffmpeg', 'win_x64'
os.chmod(FFMPEG_PATH, stat.S_IRWXU+stat.S_IRWXG+stat.S_IRWXO)
os.environ['KMP_DUPLICATE_LIB_OK'] = 'True'
# 如果开启了快速模式则强制关闭ACCURATE_MODE
if FAST_MODE:
ACCURATE_MODE = False
if SUPER_FAST:
MODE = 'FAST'
# ×××××××××××××××××××× [不要改] end ××××××××××××××××××××