公开(公告)号：CN115439712A

公开(公告)日：2022-12-06

申请号：CN202211053038.6

申请日：2022-08-30

Applicant: 南京理工大学

Inventor： 郑纯 ,  帅世康 ,  陈志华 ,  闫国旭

IPC: G06V10/774 ,  G06V10/82 ,  G06V10/44 ,  G06V10/20 ,  G06V10/25 ,  G06V10/764 ,  G06V10/766 ,  G06V20/60 ,  G06N3/04 ,  G06N3/08

Abstract: 本申请提供基于深度学习的坦克图像数据集建立及关键点检测方法，方法包括：将待检测图片输入识别模型中，检测是否存在坦克，若存在坦克输出带有边界框及关键点标签的结果图片；识别模型通过以下方法确定：建立坦克图像数据集、掩膜与关键点标注信息；批量生成坦克图像数据集，并将坦克图像数据集分为训练集，验证集与测试集；建立坦克检测模型；建立坦克关键点检测模型；对图像数据进行增强处理；分别进行坦克检测模型、坦克关键点检测模型的训练。本申请针对坦克图像数据较少的问题提出解决方法，可高效生成大量可用数据集。