发明公开
- 专利标题: 基于Transformer的跨域双分支对抗域适应图像分类方法
-
申请号: CN202310688466.4申请日: 2023-06-12
-
公开(公告)号: CN116740434A公开(公告)日: 2023-09-12
- 发明人: 潘杰 , 高洪伟 , 邹筱瑜 , 刘新华 , 顾进恒
- 申请人: 中国矿业大学
- 申请人地址: 江苏省徐州市大学路1号
- 专利权人: 中国矿业大学
- 当前专利权人: 中国矿业大学
- 当前专利权人地址: 江苏省徐州市大学路1号
- 代理机构: 南京经纬专利商标代理有限公司
- 代理商 朱小兵
- 主分类号: G06V10/764
- IPC分类号: G06V10/764 ; G06V10/80 ; G06V10/77 ; G06V10/25 ; G06V10/26 ; G06V10/82 ; G06N3/045 ; G06N3/084 ; G06N3/094
摘要:
本发明公开了一种基于Transformer的跨域双分支对抗域适应图像分类方法,包括:分别读取源域和目标域的图像;构建基于Transformer的跨域双分支对抗域适应网络模型;利用预处理后的源域和目标域图像对跨域双分支对抗域适应网络模型进行训练,将训练后的跨域双分支对抗域适应网络模型作为图像分类模型;将源域图像和目标域图像配对后分别输入至图像分类模型的双分支特征提取器中提取特征,利用分类器预测目标域中图像的标签类别。本发明将特征提取器精简为并行可交互的双分支结构,减少计算量,提升了图像训练和推理的效率;将源域和目标域的查询融合为跨域统一查询量,平滑域间分布差异,融合两个域的查询信息,用于交叉注意力在域间进行关系建模,有利于迁移知识。