我正在尝试为图像数据集训练一个多层感知器。图像尺寸为300*300,输入层为90000。这样创建是否正确?
回答:
90000是一个巨大的层,我猜你是在使用消费级设备。错误可能是Tensorflow内存不足导致的。
如果你能提供完整的错误追踪信息,我可以给出更具体的建议。
一般来说,对于基本的图像分类任务:
尝试先将图像输入到一个维度为300的卷积网络中,然后通过池化来减少空间维度。
我正在尝试为图像数据集训练一个多层感知器。图像尺寸为300*300,输入层为90000。这样创建是否正确?
回答:
90000是一个巨大的层,我猜你是在使用消费级设备。错误可能是Tensorflow内存不足导致的。
如果你能提供完整的错误追踪信息,我可以给出更具体的建议。
一般来说,对于基本的图像分类任务:
尝试先将图像输入到一个维度为300的卷积网络中,然后通过池化来减少空间维度。