Commit Graph

30 Commits

Author SHA1 Message Date
yaofanguk
bc31d1ff2d 去除propainter 2026-04-10 16:08:17 +08:00
flavioy
93d822d067 修复LAMA模式100%卡死:帧区间扩展超出视频总帧数导致FramePrefetcher死锁
- 限制字幕区间end不超过frame_count,防止内循环消费哨兵后外层永久阻塞
- LAMA批量推理改为mini-batch(4帧),避免GPU OOM
- 各inpaint模型空inpaint_area时返回原始帧
- FFmpeg子进程添加600s超时保护

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-04-08 23:34:53 +08:00
flavioy
e801d58e80 GPU加速和批处理优化、更新README
Some checks failed
Docker Build and Push / check-secrets (push) Successful in 3s
Docker Build and Push / build-and-push (cpu, latest) (push) Has been skipped
Docker Build and Push / build-and-push (cuda, 11.8) (push) Has been skipped
Docker Build and Push / build-and-push (cuda, 12.6) (push) Has been skipped
Docker Build and Push / build-and-push (cuda, 12.8) (push) Has been skipped
Docker Build and Push / build-and-push (directml, latest) (push) Has been skipped
Build Windows CPU / build (push) Has been cancelled
Build Windows CUDA 11.8 / build (push) Has been cancelled
Build Windows CUDA 12.6 / build (push) Has been cancelled
Build Windows CUDA 12.8 / build (push) Has been cancelled
Build Windows DirectML / build (push) Has been cancelled
- STTN Auto/Det: 统一 torch.no_grad 包裹,减少重复上下文切换开销
- STTN Auto: 添加 FramePrefetcher 帧预读取,根据 GPU 显存动态调整 batch size
- Lama Inpaint: 新增 _inpaint_batch 批量推理,多帧合并一次 GPU 推理
- ProPainter: copy.deepcopy 替换为浅拷贝,每个区域处理后 gc.collect
- HardwareAccelerator: 新增 get_available_vram_mb 显存查询方法
- README: 添加应用 Logo,同步英文版 README_en.md

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-04-08 00:17:50 +08:00
flavioy
6c61ddc6b0 性能优化:帧采样、FFmpeg编码、帧预读取、消除冗余拷贝
- 字幕检测:每3帧采样一次OCR,中间帧插值填充,检测速度提升约3倍
- 视频编码:cv2.VideoWriter(mp4v) 替换为 FFmpeg libx264 管道编码,画质更好、体积更小
- 帧预读取:后台线程预解码视频帧,I/O 与模型推理重叠
- 消除 deepcopy:numpy 数组改用 .copy() 替代 copy.deepcopy,降低内存开销
- 清理冗余颜色空间转换中的 np.array() 包装

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-04-07 22:58:52 +08:00
Jason
e26e23ad6a 支持设置时间选区
支持方向键快进快退(ctrl + -> or shirft + -> or ->)
2025-05-22 15:09:59 +08:00
Jason
c673b5ccd1 支持多字幕区域 2025-05-22 08:42:00 +08:00
Jason
f78e985e1c 使用PySide6-Fluent-Widgets重构整套UI
添加任务列表组件并优化视频加载逻辑
支持可视化显示字幕区域
整理所有模型, 分别为STTN智能擦除, STTN字幕检测, LAMA, ProPainter, OpenCV
提高处理性能
新增CPU运行模式并优化多语言支持
修复Propainter模式部分视频报错

本次提交新增了CPU运行模式,适用于无GPU加速的场景。同时,优化了多语言支持,新增了日语、韩语、越南语等语言配置文件,并更新了README文档以反映新的运行模式和多语言支持。此外,修复了部分代码逻辑,提升了系统的稳定性和兼容性。
2025-05-22 08:41:59 +08:00
Jason
746db4bced DirectML版本支持运行STTN模型(Windows) 2025-04-25 13:01:31 +08:00
Jason
30e7913981 修复结束时inpaint_area报错 2025-04-25 13:01:26 +08:00
Jason
87d8d9d3d7 适配torch 2.8.0 nightly build 2025-04-25 13:00:58 +08:00
天涯古巷
53baf28326 Revert "由于PySimpleGUI作者故意移除免费的旧版本,改用PySimpleGUI-4-foss" 2025-04-25 11:03:16 +08:00
Jason
97b4159d38 DirectML版本支持运行STTN模型(Windows) 2025-04-24 15:56:13 +08:00
Jason
bb80445cf4 修复结束时inpaint_area报错 2025-04-24 15:53:28 +08:00
Jason
77758d258b 适配torch 2.8.0 nightly build 2025-04-24 15:47:23 +08:00
YaoFANGUK
d6736d9206 添加sttn训练代码 2024-01-08 17:48:21 +08:00
YaoFANGUK
ba396d9569 未传入字幕区域时,进行全屏处理 2023-12-29 08:45:20 +08:00
YaoFANGUK
d25e34f621 vsr v1.1.0 2023-12-28 14:24:17 +08:00
YaoFANGUK
125a06ca50 修改config备注 2023-12-28 10:59:46 +08:00
YaoFANGUK
313c3d37a7 继续修复bug 2023-12-27 20:32:00 +08:00
YaoFANGUK
4d3d4b59bd 修复自动检测文本时若mask高度大于宽度进程卡住的bug 2023-12-27 09:22:48 +08:00
YaoFANGUK
18d57f2a18 修复bug 2023-12-26 19:12:48 +08:00
YaoFANGUK
935c341c32 修复卡住bug 2023-12-26 17:21:30 +08:00
YaoFANGUK
59cbb411af minor 2023-12-26 10:15:10 +08:00
YaoFANGUK
41a95dac0f 新增视频inpaint方法 2023-12-26 10:11:09 +08:00
天涯古巷
9aeb8f8860 Update sttn_inpaint.py 2023-12-23 10:49:30 +08:00
天涯古巷
5c819f70d1 Update sttn_inpaint.py 2023-12-23 10:42:31 +08:00
YaoFANGUK
ceb44ba034 sttn优化 2023-12-22 18:05:32 +08:00
YaoFANGUK
cf6df5040b 新增sttn 2023-12-22 12:24:43 +08:00
YaoFANGUK
29c5317a69 修复bug 2023-12-13 19:50:15 +08:00
YaoFANGUK
5e048175d3 vsr v1.0.0 2023-12-08 17:12:26 +08:00