Drowning_Person_Detection/readme.txt

15 lines
517 B
Plaintext
Raw Permalink Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

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