python - 重塑图像以适应 SVM 分类器

Question

我有这种形状的图像，(80000, 224, 224, 3)我想将这些图像拟合到 SVM 分类器，但我需要重塑它们，它会怎么做？

标签： pythonnumpymachine-learningcomputer-vision

Answer 1

假设您的形状(80000, 224, 224, 3)代表(n_samples, pixel_width, pixel_height, n_channels)，并假设您想将其重塑为 shape 的二维矩阵(n_samples, n_features)，以下代码将有所帮助。

m_samples = orig_image_matrix.shape[0]
image_matrix = orig_image_matrix.reshape(m_samples, -1)

对于带有 -1 的维度，该值是从数组的长度和剩余维度推断出来的。