使用Random Forest进行光学图像目标检测时遇数组重塑错误求助

一、ROI标签出现0、1、3三类的问题 #

可能原因 #

• Shapefile转栅格时参数错误：使用栅格化工具时未明确指定赋值，导致工具默认填充了额外值；
• 原始Shapefile拓扑异常：多边形存在重叠、无效几何，栅格化时触发异常赋值；
• 转换流程误操作：多次转换或使用第三方工具时引入了非预期的像素值。

解决步骤 #

1. 验证原始数据
   • 检查Shapefile属性表，确认所有目标多边形的标记值为1，无其他属性；
   • 用gdalinfo landuse_roi.tif查看ROI栅格的像素值统计，定位3这个值的来源。
2. 重新执行栅格化
   使用GDAL命令行重新转换，确保参数准确：

   gdal_rasterize -burn 1 -a_nodata 0 -tr [X分辨率] [Y分辨率] -te [Xmin Ymin Xmax Ymax] input.shp landuse_roi.tif
   

   -burn 1强制给多边形赋值1，-a_nodata 0指定无数据区域为0。
3. 临时修复现有ROI
   若确认3是误赋值，可通过代码重置：

   # 将非0/1的像素设为无数据（0）
   roi[~np.isin(roi, [0, 1])] = 0
   # 若3是有效数据误标记，可改为：roi[roi == 3] = 1
   

二、数组重塑时的ValueError #

错误原因 #

报错显示数组总元素数16884608与目标形状(4221152,3)的总元素数12663456不匹配。核心问题：

• 代码img[:, :, :np.int_(img.shape[2])]取了全部波段，但试图重塑为(像素数, 波段数-1)的形状，元素数量不匹配；
• 注释中提到的"Fmask波段"与实际图像的波段索引不符，未正确移除目标波段。

解决步骤 #

1. 确认图像波段信息
   先打印图像形状，明确波段数量：

   print(img.shape)  # 输出格式：(高度, 宽度, 波段数)
   

   根据报错计算，你的图像总像素数为4221152，总元素数16884608，说明波段数为4。
2. 正确移除Fmask波段
   根据训练代码中X[:, 1] <=4的逻辑，Fmask是索引为1的波段（第二个波段），移除后再重塑：

   # 删除索引为1的Fmask波段
   img_no_fmask = np.delete(img, 1, axis=2)
   # 计算新形状：(总像素数, 剩余波段数)
   new_shape = (img.shape[0] * img.shape[1], img_no_fmask.shape[2])
   img_as_array = img_no_fmask.reshape(new_shape)
   

   若Fmask是最后一个波段，可简化为：

   new_shape = (img.shape[0] * img.shape[1], img.shape[2] - 1)
   img_as_array = img[:, :, :-1].reshape(new_shape)
   

3. 验证结果
   执行后打印img_as_array.shape，确认形状符合预期（总元素数=高度×宽度×剩余波段数）。

内容的提问来源于stack exchange，提问作者Rachele Franceschini