NIMA:神经影像评估 实现 Keras + Tensorflow中的,并在AVA数据集上训练MobileNet模型的权重。 NIMA为图像分配了“均值+标准偏差”得分,可以用作自动检查图像质量的工具,也可以用作损失函数来进一步改善生成的图像的质量。 包含在AVA数据集上针对以下模型训练的权重: NASNet Mobile(由于 !而在valset上的EMD为0.067 EMD,仅需预训练即可0.0848 EMD) Inception ResNet v2(在valset上约为0.07 EMD,感谢 !) MobileNet(valset上为0.0804 EMD) 用法 评价 有eva
2022-03-29 14:25:05 14.01MB tensorflow keras ava-dataset neural-image-assessment
1
NIMA:神经影像评估 这是Hossein Talebi和Peyman Milanfar撰写的《 (在接受)的PyTorch实现。 您可以从了解更多。 实施细节 该模型在上进行了训练,该包含大约255,500张图像。 你可以从得到它。 注意:数据集中可能存在一些损坏的图像,请在开始训练之前先将其删除。 该数据集分为229,981张图像用于训练,12,691张图像用于验证和12,818张图像用于测试。 ImageNet预训练的VGG-16用作模型的基础网络,为此,我在验证集上损失了约0.072 EMD。 尚未在本文中尝试过其他两个选项(MobileNet和Inception-v2)。 非常欢迎您进行自己的扩展。 学习率设置与原始论文不同。 我似乎无法使用3e-7的转换底数和3e-6的密集区块的底线来使模型收敛于动量SGD。 另外,我没有做太多的超参数调整,因此您可能会获得更好的结果
1