
卷积神经网络能够预测图像的审美和技术价值。
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
图像质量评估本存储库依托于Google发表的研究论文,提供了美学和技术图像质量的建模方案。您可以在他们的网站上找到一份简要的概述。NIMA包含了两个模型,分别旨在预测图像的美学品质以及其技术质量。为了提升模型的性能,我们采用了迁移学习的方法,利用ImageNet预训练的卷积神经网络(CNN)进行分类任务并对其进行精细调整。关于如何将NIMA应用于解决具体问题,我们在两个技术博客文章中详细阐述了相关内容:NVIDIA开发者博客提供了可用的代码,允许您使用任何预先训练好的模型。此外,我们还提供了Docker镜像,用于在AWS EC2平台上进行本地CPU训练以及远程GPU训练,同时提供了与数据集配合使用的预训练模型。完整的文档可以在以下链接中查阅: 。图像质量评估与Python 3.6兼容,并遵循Apache 2.0许可协议进行分发。我们热忱欢迎各种形式的贡献,特别是那些能够改进当前发布模型性能的新型架构和/或超参数组合(请参考 )。这些模型涵盖了对审美模型的预测以及对技术模型的预测。我们提供经过充分训练的模型,无论是美学分类还是技术分类,都基于MobileNet作为基础CNN架构。这些模型及其对应的配置文件均存储在models/MobileNet 目录下,并且展现出良好的性能表现。
全部评论 (0)
还没有任何评论哟~


