
Neural-IMage-Assessment:基于PyTorch的神经影像评估实现
5星
- 浏览量: 0
- 大小:None
- 文件类型:ZIP
简介:
Neural-IMage-Assessment是一款利用PyTorch框架构建的工具,专注于通过深度学习技术进行高效的神经影像分析与评估。
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。此外,我没有进行过多的超参数调整,因此您可能会获得更好的结果。
全部评论 (0)
还没有任何评论哟~


