3.28日报

在系统集成上也进行了创新

def process_frame(vis_frame, ir_frame):
    """创新点5:实时处理流水线"""
    start_time = time.time()
    
    # 烟雾处理
    with torch.no_grad():
        dehazed_frame, density_map = dehaze_model(vis_frame)
    
    # 人体检测
    with torch.no_grad():
        detections = detection_model(dehazed_frame, ir_frame)
    
    # 后处理
    processed_results = post_process(detections)
    
    end_time = time.time()
    processing_time = end_time - start_time
    
    return {
        "dehazed_image": dehazed_frame,
        "density_map": density_map,
        "detections": processed_results,
        "processing_time": processing_time
    }

 

posted @ 2025-04-09 11:09  Code13  阅读(8)  评论(0)    收藏  举报