0
点赞
收藏
分享

微信扫一扫

opencv--使用直方图找谷底进行确定分割阈值

直方图原理就不说了,大家自行百度

直方图可以帮助分析图像中的灰度变化,进而帮助确定最优二值化的灰度阈值(threshold level)。如果物体与背景的灰度值对比明显,此时灰度直方图就会包含双峰(bimodal histogram),即直方图中一般会有两个峰值,分别为图像的前景和背景。

前景使得某个灰度区间的灰度值的数量急剧增加,就会产生一个峰值,同理背景会使另一个灰度区间的灰度值的数量急剧增加,就产生另外一个峰值,两峰间的谷底对应于物体边缘附近相对较少数目的像素点。

这两个峰值之间的最小值一般就是最优二值化的分界点,通过这个分界点可以把前景和背景很好地分割开来。

有时这两个峰值会有部分重叠,即左侧峰值的下降部分和右侧峰值的上升部分存在叠加。通常可以把自然界的信号看做高斯信号,即一个峰值对应一个高斯信号,当直方图中的两个高斯信号在某个灰度区域叠加的时候,其叠加区就形成了一个圆滑的谷底,就很难找到一个确切的位置(最优二值化的灰度值)把这两个峰值分开。

 

float calculateThreshold(cv::Mat& img)
{
    cv::Mat temp = img.clone();
    
    // 计算直方图
    cv::Mat hist;
    int histSize = 256;  // 直方图尺寸
    float range[] = { 0, 256 };  // 像素值范围
    const float* ranges[] = { range };
    cv::calcHist(&img, 1, nullptr, cv::Mat(), hist, 1, &histSize, ranges);

    /*for (int i = 0; i < 21; i++)
        hist.at<float>(i, 0) = 0.0;*/

    cv::normalize(hist, hist, 0, 1, cv::NORM_MINMAX);
    //hist.convertTo(hist, CV_32S);
    cv::GaussianBlur(hist, hist, cv::Size(0, 0),3,3);
    //cv::blur(hist, hist, cv::Size(1, 9),cv::Point(-1,-1));
    
    
    
    std::vector<float> peaks;  // 存储峰值位置
    std::vector<float> valleys;  // 存储低谷位置

    for (int i = 1; i < histSize - 1; i++) {
        //std::cout << std::fixed << std::setprecision(4);
        float currentValue = hist.at<float>(i);
        float prevValue = hist.at<float>(i - 1);
        float nextValue = hist.at<float>(i + 1);
        
        /*if (currentValue < 0.001)
            continue;*/
        // 具体情况需要修改currentValue>0.005的阈值
        if ((currentValue > prevValue && currentValue > nextValue && currentValue>0.005)) {
            std::cout << prevValue << " " << currentValue << " " << nextValue << std::endl;
            peaks.push_back(i);  // 峰值
        }
        else if (currentValue < prevValue && currentValue < nextValue && currentValue>0.001) {
            std::cout << prevValue << " " << currentValue << " " << nextValue << std::endl;
            valleys.push_back(i);  // 低谷
        }
    }
    if(valleys.size()>0)
        cv::threshold(temp, temp, valleys[0], 255, cv::THRESH_BINARY);
    // 创建直方图可视化图像
    int histWidth = 512;
    int histHeight = 400;
    cv::Mat histImage(histHeight, histWidth, CV_8UC3, cv::Scalar(0, 0, 0));
    cv::Mat hist_temp;
    // 归一化直方图数据
    cv::normalize(hist, hist_temp, 0, histImage.rows, cv::NORM_MINMAX, -1, cv::Mat());
    
    // 绘制直方图
    int binWidth = cvRound((double)histWidth / histSize);
    for (int i = 0; i < histSize; i++) {
        int binHeight = cvRound(hist_temp.at<float>(i));
        cv::line(histImage, cv::Point(i * binWidth, histHeight), cv::Point(i * binWidth, histHeight - binHeight), cv::Scalar(255, 255, 255));
    }

    if (valleys.size() > 0)
        return valleys[0];
    return 0;
}

举报

相关推荐

0 条评论