learn opencv-如何在OpenCV中选择边界框(ROI)?

参考:https://github.com/spmallick/learnopencv


如何在OpenCV(C ++ / Python)中选择边界框(ROI)?

在本教程中,我们将学习如何在OpenCV中的图像中选择边界框或感兴趣区域(ROI)。 在过去,我们必须通过处理鼠标事件来编写我们自己的边界框选择器。 但是,现在我们可以选择使用OpenCV原生部分的函数selectROI。

我对OpenCV库中的奇怪选择感到惊讶。 您可能会认为selectROI将成为具有显示图像,绘图等功能的highgui的一部分。但是,selectROI是跟踪API的一部分! 正如你在后面会注意到的,写selectROI时所做的选择有点奇怪。 但是,在我们批评之前,我们得感谢有人制造出一些有用的东西,尽管它并不完美。

让我们深入看看selectROI的用法

如何在OpenCV中选择一个感兴趣的区域

由于selectROI是跟踪API的一部分,因此您需要使用opencv_contrib安装OpenCV 3.0(或更高版本)。

我们从一个示例代码开始。 它允许您在图像中选择一个矩形,裁剪矩形区域并最终显示裁剪后的图像。

我们将修改突出显示的行以尝试不同的选项。

C++

#include 
// selectROI is part of tracking API
#include 

using namespace std;
using namespace cv;


int main (int argc, char **arv)
{
    // Read image
    Mat im = imread("image.jpg");

    // Select ROI
    Rect2d r = selectROI(im);

    // Crop image
    Mat imCrop = im(r);

    // Display Cropped Image
    imshow("Image", imCrop);
    waitKey(0);

    return 0;
}

Python

import cv2
import numpy as np

if __name__ == '__main__' :

    # Read image
    im = cv2.imread("image.jpg")

    # Select ROI
    r = cv2.selectROI(im)

    # Crop image
    imCrop = im[int(r[1]):int(r[1]+r[3]), int(r[0]):int(r[0]+r[2])]

    # Display cropped image
    cv2.imshow("Image", imCrop)
    cv2.waitKey(0)

从左上角拖动矩形到右下角

如果你像我一样,你会喜欢把一个矩形从左上角拖到右下角,而不是从中心拖动。 那么,我们可以很容易地解决这个问题,用下面一行代替突出显示的行。

C++

bool fromCenter = false; 
Rect2d r = selectROI(im, fromCenter);

Python

fromCenter = False
r = cv2.selectROI(im, fromCenter)

如何指定窗口名称?

如果你可以使用现有的窗口而不是ROI选择器的窗口,这不是很好吗? 那么你走了

C++

bool fromCenter = false; 
Rect2d r = selectROI("Image", im, fromCenter);

Python

fromCenter = False
r = cv2.selectROI("Image", im, fromCenter)

不想显示十字线?

现在,假设你不喜欢十字准线,并希望看到没有它的矩形。 您可以修改代码以不显示十字线。

C++

bool showCrosshair = false; 
bool fromCenter = false; 
Rect2d r = selectROI("Image", im, fromCenter, showCrosshair);

Python

showCrosshair = False
fromCenter = False
r = cv2.selectROI("Image", im, fromCenter, showCrosshair)

如何在一个图像上选择多个感兴趣区域(roi)?

selectROI函数还允许您选择多个感兴趣的区域,但似乎有两个错误。

Bug警告1:根据说明,您可以拖动一个矩形,然后按ENTER并拖动另一个矩形。 但是,OpenCV 3.2中的实现似乎存在一个错误。 您必须在第一个矩形后两次按ENTER键。 对于所有后续的矩形,您应该按ENTER一次。

C++

// Specify a vector of rectangles (ROI) 
vector rects;
bool fromCenter = false; 
// The selected rectangles are in
selectROI("Image", im, rects, fromCenter);

错误警报2:我不能让python版本工作,没有文档。 下面的代码运行,但是变量rects没有被填充。 该函数也不会返回任何东西。 如果您发现修复,请在下面的评论中告诉我。

Python

# Note this code does not work. 
# Specify a vector of rectangles (ROI) 
rects = []
fromCenter = false
# Select multiple rectangles
selectROI("Image", im, rects, fromCenter)

你可能感兴趣的:(opencv)