一种基于PaddleDetection的混淆矩阵生成方法

    公开(公告)号:CN114120057A

    公开(公告)日:2022-03-01

    申请号:CN202111320510.3

    申请日:2021-11-09

    Applicant: 华侨大学

    Abstract: 本发明提供了一种基于PaddleDetection的混淆矩阵生成方法,包括:在PaddlePaddle平台下运用PaddleDetection目标检测开发套件训练目标检测网络;获取有标注的测试数据集,分为两个文件夹,一个文件夹放json文件,另一个放对应的图片文件,并得到真实的目标物体外接矩形框;运用infer.py图片文件进行预测,得到预测标签;得出预测修正得到的矩形检测框;将真实矩形框与预测修正得到的矩形检测框进行循环匹配,将统计得到的值记录在矩阵上;创建excle表格,将记录的矩阵填充到excle表格中,绘制出用于评价模型实际应用效果的混淆矩阵,本发明方法能通过目标检测网络预测的检测框与实际的目标检测框进行匹配,绘制在混淆矩阵上,以便对于模型各个种类的识别结果进行统计研究,优化网络模型的训练方法。

    一种实例分割模型训练方法、装置及可读介质

    公开(公告)号:CN115601546A

    公开(公告)日:2023-01-13

    申请号:CN202211324514.3

    申请日:2022-10-27

    Abstract: 本发明公开了一种实例分割模型训练方法、装置及可读介质,通过采用第一训练集分别训练主体模型和辅助模型,得到经训练的主体模型和经训练的辅助模型,其中,主体模型为实例分割模型,辅助模型为逐像素分类模型;将待测图像分别输入经训练的主体模型和经训练的辅助模型,得到第一预测结果和第二预测结果,分别获取第一预测结果和第二预测结果的第一掩膜图和第二掩膜图;根据第一掩膜图和第二掩膜图确定难例样本,难例样本为因置信度低而未被经训练的主体模型识别出目标的待测图像,将难例样本与其第二预测结果对应的标签保存为第二训练集;采用第二训练集对经训练的主体模型继续进行训练,得到训练好的主体模型,能够有效提高模型泛化性和准确率。

Patent Agency Ranking