二抽取代码MATLAB-rnn-biqa:基于RNN的盲图质量模型

时间:2021-05-26 12:30:42
【文件属性】:
文件名称:二抽取代码MATLAB-rnn-biqa:基于RNN的盲图质量模型
文件大小:5KB
文件格式:ZIP
更新时间:2021-05-26 12:30:42
系统开源 二摘代码MATLAB 基于RNN的盲图质量模型(RNN-BIQA) RNN-BIQA模型将输入图像分为补丁,然后使用CNN模型从每个补丁中提取特征,以形成特征向量序列。 然后,基于RNN的模型可用于从特征向量序列中预测主观质量得分(MOS)。 训练模型包括两个阶段。 首先,需要在训练集中的图像块上训练用于特征提取的CNN模型。 我们提供了经过预训练的CNN特征提取器。 有关如何训练CNN特征提取器的建议,请参考CNN-TLVQM视频质量模型的GitHub页面。 请注意,CNN-TLVQM和RNN-BIQA中使用的CNN模型相似,但不完全相同。 对于第二阶段,训练和测试基于RNN的质量模型,您可以参考KoNIQ_Example.mat中的Matlab代码示例。 要使用此示例,您需要下载KoNIQ-10k图像质量,可在中获得。 该代码已使用Matlab R2020a版本进行了测试。 请注意,Matlab的早期版本不支持GRU层。
【文件预览】:
rnn-biqa-main
----huberRegressionLayer.m(2KB)
----README.md(1KB)
----MakeRNNFeatures.m(4KB)
----TrainRNNModel.m(2KB)
----KoNIQ_Example.m(2KB)

网友评论