基于Python的无参考图像质量评价

源码下载地址:https://download.csdn.net/download/sheziqiong/85602575
源码下载地址:https://download.csdn.net/download/sheziqiong/85602575

无参考图像质量评价

简介

参考RankIQA中的方法,将回归问题转化为分类+回归的
问题,对高质量图像进行distortion,生成大量不同等级的图像,先训练一个图像质量rank的网络,
网络结构为Siamese,Siamese内为VGG;在训练完成后,在其基础上进行fine-tuning,使用公开集
数据(大约为2000多张)微调该网络,损失函数为L2,回归图像的具体质量分数。

该方法可以突破无参考图像质量评价的小数据集限制,可以使用大网络来训练而不产生过拟合。
在这里插入图片描述

文件说明

  • ./data: distortion的matlab代码,生成Rank网络需要的不同等级的图像
  • src: 为适应人脸大小,将网络输入从224降低到128重新训练
  • _src: 原网络代码,输入大小为224
  • regression_network: 使用小网络拟合RankIQA的效果
    • train_pose_qua.py: 一个网络同时完成预测人脸角度和图像质量的任务
      • 先使用该网络分别完成单个任务,观察最终loss可以下降到多少,这个值是这个网络性能的极限
      • 防止一个任务的loss被另一个任务影响,需要将两个任务的loss加权,当作反向传播的loss
      • 分别打印出两个任务的loss,观察是否下降到单任务时的最小loss
      • 训练时发现在低质量图像时网络性能很差,对图像的质量label进行非线性拉伸到 0-10,使用的非线性www.biyezuopin.vip
        拉伸公式为 qua_final = qua0.4*(10./5.50.4) (label中的最大值为5.5)

源码下载地址:https://download.csdn.net/download/sheziqiong/85602575
源码下载地址:https://download.csdn.net/download/sheziqiong/85602575

  • 1
    点赞
  • 30
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值