如何使用open cv在Brain肿瘤上绘制矩形？

Question

我想在我的脑肿瘤图像数据集上绘制绿色的矩形或边界框，以突出图像中的肿瘤。什么是最短的方法呢？我不想分割肿瘤。

我试过openev绘制矩形函数，但它不起作用。我也发现了Active Contour和Canny算法，但这也行不通。

    import cv2 
    import numpy as np 
    image = cv2.imread('12.png')
    cv2.imshow('Input Image', image) 
    cv2.waitKey(0) 
    gray =cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

    # Canny Edges edged = cv2.Canny(gray, 30, 200) 
    # cv2.imshow('Canny Edges', edged) 
    # cv2.waitKey(0)

    _, contours, hierarchy = cv2.findContours(edged, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_NONE) 
    cv2.imshow('Canny Edges after Contouring',edged) 
    cv2.waitKey(0) 
    print("Number of contours found = " +str(len(contours))) 
    cv2.drawContours(image, contours, -1, (0, 255, 0),3) 
    cv2.imshow('Contours', image) 
    cv2.waitKey(0)
    cv2.destroyAllWindows()

我希望肿瘤部分应该是整个图像的一个盒子。图像区域的恢复应该与我的实际脑部图像相同。只需要一个绿色的盒子，通过简单的代码而不是整个算法。

Answer 1

根据对这个答案的评论，我更新了我的答案，我认为这仍然是一个深度学习问题，但不同类型。

如果你需要沿肿瘤的轮廓分割肿瘤，一种方法可能就是Mask R-CNN（1）（你可以阅读2017年发表的原始论文）。该方法允许沿其轮廓进行图像分割。 Here你可以找到该方法的示例实现。

从图像中我可以看出，你必须处理灰度图像，并且可能很难通过简单的方法从场景的其余部分分割肿瘤，这里我同意@BahramdunAdil。因此，我怀疑培训可能需要大量的数据和时间。

祝好运！

（1）何凯明等。 “面具r-cnn。” IEEE计算机视觉国际会议论文集。 2017年。