main函数: 配置参数、加载模型、图像测试。 AI_process函数: '''对原图进行目标检测和水域分割''' '''输入:检测模型、分割模型、配置参数、路径 返回:返回目标检测结果、原图像、分割图像, ''' AI_postprocess函数: '''根据预测和分割结果,对是否为落水人员进一步后处理''' '''输入:落水人员的结果(类别+坐标)、原图、mask图像 过程:获得mask的轮廓,判断人员是否在轮廓内。 在,则保留且绘制;不在,舍弃。 返回:最终绘制的结果图、最终落水人员(坐标、类别、置信度), '''