Sobel算子处理图像出现截断拼接异常，求解决方案

输入为8位灰度BMP图像，输出图像出现截断部分移至开头的异常现象，以下是问题代码及解决思路：

问题代码 #

int sobeloperator(const char* inputimage, const char* outputimage)
{
    const int Gx[3][3] = {
        {-1, 0, 1},
        {-2, 0, 2},
        {-1, 0, 1}
    };

    const int Gy[3][3] = {
        {-1, -2, -1},
        {0, 0, 0},
        {1, 2, 1}
    };

    BMPHeader header;
    BMPInfoHeader infoHeader;
    uint8_t* inputPixels;
    uint8_t* outputPixels;

    ifstream inFile(inputimage, ios::in | ios::binary);
    if (!inFile.is_open()) {
        cerr << "Error: Unable to open input file." << endl;
        return 1;
    }

    // Read BMP header

    inFile.read(reinterpret_cast<char*>(&header), sizeof(header));
    if (header.type != 0x4D42) {
        cerr << "Error: Invalid BMP file format." << endl;
        return 1;
    }

    // Read BMP info header

    inFile.read(reinterpret_cast<char*>(&infoHeader), sizeof(infoHeader));

    if (infoHeader.bpp != 8) {
        cerr << "Error: Only 8-bit BMP images are supported." << endl;
        return 1;
    }

    char colors[1024] = { 0 };
    for (int i = 0; i < 256; i++)
    {
        colors[i * 4] = (char)i;
        colors[i * 4 + 1] = (char)i;
        colors[i * 4 + 2] = (char)i;
    }

    inputPixels = new uint8_t[infoHeader.width * infoHeader.height];
    outputPixels = new uint8_t[infoHeader.width * infoHeader.height];

    inFile.read(reinterpret_cast<char*>(inputPixels), infoHeader.width * infoHeader.height);

    for (int y = 0; y < infoHeader.height; y++)
    { 
        for (int x = 0; x < infoHeader.width; x++)
        {
            int GxSum = 0;
            int GySum = 0;
            for (int k = -1; k <= 1; k++) {
                for (int l = -1; l <= 1; l++) {
                    // get pixel value at current kernel position
                    int pos = (y + k) * infoHeader.width + (x + l); //current pos, infoHeader.width is a row, multiplied by y+k = current row.

                    if (pos < 0 || pos >= infoHeader.width * infoHeader.height || y <= 0 || y >= infoHeader.height || x <= 0 || x >= infoHeader.width) {
                        continue;
                    }

                    // add pixel value multiplied by Sobel kernel value to Gx and Gy sums
                    GxSum += Gx[k+1][l+1] * inputPixels[pos];
                    GySum += Gy[k+1][l+1] * inputPixels[pos];
                }
            }
            int gradientMagnitude = sqrt(GxSum * GxSum + GySum * GySum);

            // clip gradient magnitude to 8-bit range
            if (gradientMagnitude <= 128)
            {
                gradientMagnitude = 0;
            }
            else
            {
                gradientMagnitude = 255;
            }
            //gradientMagnitude = min(255, max(0, gradientMagnitude));
            
            // set output pixel to gradient magnitude
            
            outputPixels[y * infoHeader.width + x] = gradientMagnitude;
        }

    }
    
    infoHeader.sizeImage = infoHeader.width * infoHeader.height;
    header.size = sizeof(header) + sizeof(infoHeader) + sizeof(colors) + infoHeader.sizeImage;
    header.offset = sizeof(header) + sizeof(infoHeader) + sizeof(colors);
    printf("

bpp = %d
", infoHeader.bpp);
    printf("compression = %d", infoHeader.compression);
    // Write output BMP file
    ofstream outFile("sobel.bmp", ios::out | ios::binary);
    if (!outFile.is_open()) {
        cerr << "Error: Unable to create output file." << endl;
        return 1;
    }

    outFile.write(reinterpret_cast<char*>(&header), sizeof(header));
    outFile.write(reinterpret_cast<char*>(&infoHeader), sizeof(infoHeader));
    outFile.write(colors, 1024);
    outFile.write(reinterpret_cast<char*> (outputPixels), infoHeader.sizeImage);
    delete[] inputPixels;
    delete[] outputPixels;
    inFile.close();
    outFile.close();


    return 0;

}

解决思路 #

1. 修复BMP行对齐问题（核心原因） #

BMP格式强制要求每行字节数为4的整数倍，不足部分用填充字节补全。代码直接按width*height读取像素，忽略填充字节导致像素数据错位，这是图像截断偏移的根源：

• 计算每行实际字节跨度：

  int rowStride = (infoHeader.width * infoHeader.bpp + 31) / 32 * 4;
  

• 逐行读取像素并跳过填充字节：

  inputPixels = new uint8_t[infoHeader.width * infoHeader.height];
  for (int y = 0; y < infoHeader.height; y++) {
      inFile.read(reinterpret_cast<char*>(&inputPixels[y * infoHeader.width]), infoHeader.width);
      inFile.seekg(rowStride - infoHeader.width, ios::cur); // 跳过填充
  }
  

• 写入输出图像时补充填充字节：

  int outputRowStride = (infoHeader.width * infoHeader.bpp + 31) / 32 * 4;
  int padding = outputRowStride - infoHeader.width;
  uint8_t pad = 0;
  for (int y = 0; y < infoHeader.height; y++) {
      outFile.write(reinterpret_cast<char*>(&outputPixels[y * infoHeader.width]), infoHeader.width);
      if (padding > 0) {
          outFile.write(reinterpret_cast<char*>(&pad), padding);
      }
  }
  

2. 修正边界判断逻辑 #

当前判断条件错误地检查了当前x/y而非偏移后的坐标，导致边界处理失效：

// 替换原边界判断代码
int ny = y + k;
int nx = x + l;
if (ny < 0 || ny >= infoHeader.height || nx < 0 || nx >= infoHeader.width) {
    continue;
}
int pos = ny * infoHeader.width + nx;

3. 优化梯度计算精度 #

sqrt返回double类型，直接转int会丢失精度，建议显式转换并取整：

double mag = sqrt(static_cast<double>(GxSum*GxSum + GySum*GySum));
int gradientMagnitude = static_cast<int>(round(mag));

4. 修正BMP头信息 #

• 用实际行跨度计算图像大小：

  infoHeader.sizeImage = outputRowStride * infoHeader.height;
  

• 确保header.size和header.offset与实际写入的文件结构一致。

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