fix(ceus): 降低长视频内存峰值,快照开跑即写,build 核对 CEUS 运行与交付包
- 内存:prepare_segmentation 改为流式解码,只留分屏和配准的抽样帧,绿框逐帧在两半上找; 分割那一半由 SegmentationInput.target_frames() 到用时再解码,segment_video 边读边写 jpg; MedSAM2 init_state 的读帧换成直接在存放设备上建张量(不改 MedSAM2 源码,结果逐位相同); 传播结果写进预分配数组。帧数最多的一段视频常驻内存峰值 11.2 GiB → 2.3 GiB,分辨率最大的 一段原代码被 OOM 杀掉、现为 4.2 GiB;两段的掩码和全部产物与改前逐字节相同,33 段视频的 第 1–2 步结果与改前一致,29 例配准与 review.csv 一致。 - 快照:ceus_quant 开跑时写 snapshot/(git、ceus.yaml、review.csv、uv.lock、segment.json), 配置和核对表从快照副本读;收尾由新函数 runs.finish_snapshot 只新建 finished.json。 build 与 ceus_align_review 同样改用 finish_snapshot 记结束时间。 - build:summary.json 加 video_sha256;装 CEUS 结果时核对快照里的 delivery_dir、逐例核对 视频 SHA-256 和推注时间,不同就整次停下;新字段做类型检查,不进 tic.json。 来源:主会话任务(问题 1–3 及补充的 delivery_dir 比对)。 Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
This commit is contained in:
@@ -72,6 +72,21 @@ def test_green_box_in_right_half(ceus_config: CeusConfig) -> None:
|
||||
assert abs(int(xs.max()) - 169) <= 3
|
||||
|
||||
|
||||
def test_frames_are_iterated_once(ceus_config: CeusConfig) -> None:
|
||||
"""帧只遍历一遍:传逐帧交出的生成器,结果与传列表相同。"""
|
||||
frames = _frames([250])
|
||||
_, left, right = split_views(frames, ceus_config.views)
|
||||
from_list = locate_contrast_side(
|
||||
frames, left, right, ceus_config.prompt_source, ceus_config.keyframes
|
||||
)
|
||||
from_generator = locate_contrast_side(
|
||||
(frame for frame in frames), left, right, ceus_config.prompt_source, ceus_config.keyframes
|
||||
)
|
||||
assert from_generator.side == from_list.side == "right"
|
||||
assert [p.frame_index for p in from_generator.prompt_frames] == [0]
|
||||
assert np.array_equal(from_generator.prompt_frames[0].mask, from_list.prompt_frames[0].mask)
|
||||
|
||||
|
||||
def test_green_box_in_both_halves_fails_case(ceus_config: CeusConfig) -> None:
|
||||
frames = _frames([40, 250])
|
||||
_, left, right = split_views(frames, ceus_config.views)
|
||||
|
||||
Reference in New Issue
Block a user