识别柠檬上的黑点

Question

我尝试了几次尝试来识别柠檬上的黑点。我无法区分黑色阴影和柠檬上的实际黑色污渍。

我尝试使用 InRange 并将图像转换为 HSV 但没有成功，老实说我很迷茫，希望能有一些新的想法来识别黑色污渍。

这是我的代码：

import cv2
import matplotlib.pyplot as plt
import numpy as np

img = cv2.imread("./data/lemon1big.jpg")
gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
ret, thresh = cv2.threshold(gray,150,255,cv2.THRESH_BINARY_INV+cv2.THRESH_OTSU)

plt.imshow(thresh)

结果：

这些是我要检测的污渍 - 我检测到 12 个污渍：

Answer 1

我建议使用自适应阈值而不是otsu，因为黑色背景打乱了otsu所做的阈值计算，然后您可以使用连接组件分析和按大小过滤来获得黑点，这里的代码：

import cv2
import matplotlib.pyplot as plt

def plotImg(img):
    if len(img.shape) == 2:
        plt.imshow(img, cmap='gray')
        plt.show()
    else:
        plt.imshow(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
        plt.show()

img = cv2.imread('lemon.png')
gray_img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
binary_img = cv2.adaptiveThreshold(gray_img, 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C,
                                   cv2.THRESH_BINARY_INV, 131, 15)
plotImg(binary_img)
_, _, boxes, _ = cv2.connectedComponentsWithStats(binary_img)
# first box is the background
boxes = boxes[1:]
filtered_boxes = []
for x,y,w,h,pixels in boxes:
    if pixels < 10000 and h < 200 and w < 200 and h > 10 and w > 10:
        filtered_boxes.append((x,y,w,h))

for x,y,w,h in filtered_boxes:
    cv2.rectangle(img, (x,y), (x+w,y+h), (0,0,255),2)

plotImg(img)

binary image

image recognized

识别柠檬上的黑点

Identify black dots on lemon

python

opencv

threshold