公开(公告)号：CN115115830A

公开(公告)日：2022-09-27

申请号：CN202210537811.X

申请日：2022-05-17

Applicant: 西北农林科技大学

Inventor： 唐晶磊 ,  丁亚伟 ,  邵嘉燕 ,  史怀平 ,  霍迎秋

IPC: G06V10/26 ,  G06V10/52 ,  G06V10/764 ,  G06V10/80 ,  G06V10/82 ,  G06V20/40 ,  G06V40/20 ,  G06N3/04 ,  G06N3/08

Abstract: 本发明涉及一种基于改进Transformer的家畜图像实例分割方法，包括以下步骤：步骤S1：获取高质量的家畜图像，进行标注和图像数据扩增，构建训练集；步骤S2：基于多尺度可变形注意力模块和统一查询表示模块对目标检测网络DETR进行改进，从而构建基于改进Transformer的家畜图像实例分割模型；步骤S3：根据训练集对基于Transformer的家畜图像实例分割模型进行训练，得到训练好的分割模型；步骤S4：根据训练好的分割模型对待检测家畜图像数据进行处理，获得实例分割效果。本发明可以有效地解决原始Transformer结构直接用于家畜图像分割时模型训练比较困难、分割性能不佳的问题。