落水人员检测
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

517B

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