图像质量神经网络项目

我正在尝试创建一个神经网络项目,用于判断彩色图像的质量,并返回该图像是否通过质量检查。这个质量检查将基于2000个样本进行训练,其中包含50%的优质照片和50%的劣质照片。

假设我正在检查的图片尺寸为1280×1280像素,采用RGB格式且无Alpha通道。我试图创建以下神经网络结构:

输入层 ==> 1280 * 1280 * 3 = 4915200个神经元

2个隐藏层 ==> 第一层 = 1638400个神经元,第二层 = 409600个神经元

输出层 ==> 1个神经元

我遇到的问题是,在简单的XOR训练中,我们可以轻松地使用2个输入、1个包含2个神经元的隐藏层和1个输出神经元来完成。然而,当应用于现实世界时,内存成为了一个巨大的问题。所需的内存量超过了我的128GB内存机器的容量。

我不知道应该如何处理这个项目才能使其在内存使用上更加高效?(至少在128GB内)

期待就我应该采取什么方法来使其工作展开公开讨论!

谢谢


回答:

你应该使用卷积神经网络。我建议你从一个图像识别项目中复制一个神经网络,并在你的数据集上重新训练。一个高效的卷积神经网络实现不应该占用那么多内存,尽管训练可能需要很长时间。

你还可以使用较低分辨率的图像,这是机器视觉中通常采用的做法。将所有图像缩小到例如256×256像素。

Related Posts

L1-L2正则化的不同系数

我想对网络的权重同时应用L1和L2正则化。然而,我找不…

使用scikit-learn的无监督方法将列表分类成不同组别,有没有办法?

我有一系列实例,每个实例都有一份列表,代表它所遵循的不…

f1_score metric in lightgbm

我想使用自定义指标f1_score来训练一个lgb模型…

通过相关系数矩阵进行特征选择

我在测试不同的算法时,如逻辑回归、高斯朴素贝叶斯、随机…

可以将机器学习库用于流式输入和输出吗?

已关闭。此问题需要更加聚焦。目前不接受回答。 想要改进…

在TensorFlow中,queue.dequeue_up_to()方法的用途是什么?

我对这个方法感到非常困惑,特别是当我发现这个令人费解的…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注