发明公开
- 专利标题: 考虑标签信息的多任务对比学习图像美学风格分类方法
-
申请号: CN202310690405.1申请日: 2023-06-12
-
公开(公告)号: CN116664945A公开(公告)日: 2023-08-29
- 发明人: 张桦 , 罗逸章 , 张灵均 , 叶挺聪 , 许艳萍 , 吴以凡 , 王慕薇
- 申请人: 杭州电子科技大学
- 申请人地址: 浙江省杭州市下沙高教园区2号大街
- 专利权人: 杭州电子科技大学
- 当前专利权人: 杭州电子科技大学
- 当前专利权人地址: 浙江省杭州市下沙高教园区2号大街
- 代理机构: 杭州君度专利代理事务所
- 代理商 朱月芬
- 主分类号: G06V10/764
- IPC分类号: G06V10/764 ; G06V10/774 ; G06V10/82 ; G06V10/40 ; G06N3/0464 ; G06N3/0895
摘要:
本发明公开了考虑标签信息的多任务对比学习图像美学风格分类方法,首先对图像美学风格数据集进行预处理;然后针对图像美学风格属性设计不同的对比学习任务,构建基于多任务对比学习的美学风格特征学习神经网络模型,使用经过预处理图像美学风格数据集对美学风格特征学习模型进行预训练,训练过程中不断调整多任务损失权重使模型可以更好地学习图像美学风格特征;预训练完成后,提取出预训练完成的美学风格特征学习模型,并连接一个风格分类器,建立图像美学风格分类模型。本发明采用了多个对比学习任务对模型进行训练,并结合多任务权重均衡策略调整任务权重,综合提升图像美学风格特征编码器的学习能力和风格分类的准确度。