对CHM模型进行分水岭单木分割的C++代码

以下是对CHM模型进行分水岭单木分割的C++代码：

#include <iostream>
#include <opencv2/opencv.hpp>

using namespace std;
using namespace cv;

int main(int argc, char** argv) {
    // 读取输入图像
    Mat srcImg = imread("input.png", IMREAD_GRAYSCALE);
    if (srcImg.empty()) {
        cerr << "Failed to read input image!" << endl;
        return -1;
    }

    // 计算梯度图像
    Mat gradX, gradY;
    Sobel(srcImg, gradX, CV_32F, 1, 0);
    Sobel(srcImg, gradY, CV_32F, 0, 1);
    Mat gradMag = magnitude(gradX, gradY);

    // 计算分水岭掩模
    Mat markers;
    threshold(gradMag, markers, 0, 255, THRESH_BINARY | THRESH_OTSU);
    markers.convertTo(markers, CV_32S);
    watershed(srcImg, markers);

    // 提取单木区域
    Mat dstImg = Mat::zeros(srcImg.size(), CV_8U);
    for (int i = 0; i < markers.rows; i++) {
        for (int j = 0; j < markers.cols; j++) {
            if (markers.at<int>(i, j) == -1) {
                dstImg.at<uchar>(i, j) = 255;
            }
        }
    }

    // 显示输出图像
    imshow("Output Image", dstImg);
    waitKey(0);
    return 0;
}

需要注意的是，该代码使用了OpenCV库，因此需要在编译时链接该库。可以使用以下命令进行编译：

g++ watershed.cpp -o watershed `pkg-config --cflags --libs opencv`