2017-06-01 79 views
0

假设我想检测图像中是否存在果酱罐。例如。在下表中,我在桌子上还有一个果酱罐。代码将检测到图像有果酱罐。如果图像中没有果酱罐,代码将突出显示,没有图像。使用OpenCV进行图像检测

我想用python中的openCV创建一个代码来检测图像。

我发现“模板匹配”是一种方法。我正在使用的代码如下:

import cv2 
import numpy as np 
from matplotlib import pyplot as plt 
img = cv2.imread('flower.jpg',0) 
img2 = img.copy() 
template = cv2.imread('jam_image.jpg',0) 
w, h = template.shape[::-1] 
# All the 6 methods for comparison in a list 
methods = ['cv2.TM_CCOEFF', 'cv2.TM_CCOEFF_NORMED', 'cv2.TM_CCORR', 
      'cv2.TM_CCORR_NORMED', 'cv2.TM_SQDIFF', 'cv2.TM_SQDIFF_NORMED'] 
for meth in methods: 
    img = img2.copy() 
    method = eval(meth) 
    # Apply template Matching 
    res = cv2.matchTemplate(img,template,method) 
    min_val, max_val, min_loc, max_loc = cv2.minMaxLoc(res) 
    # If the method is TM_SQDIFF or TM_SQDIFF_NORMED, take minimum 
    if method in [cv2.TM_SQDIFF, cv2.TM_SQDIFF_NORMED]: 
     top_left = min_loc 
    else: 
     top_left = max_loc 
    bottom_right = (top_left[0] + w, top_left[1] + h) 
    cv2.rectangle(img,top_left, bottom_right, 255, 2) 
    plt.subplot(121),plt.imshow(res,cmap = 'gray') 
    plt.title('Matching Result'), plt.xticks([]), plt.yticks([]) 
    plt.subplot(122),plt.imshow(img,cmap = 'gray') 
    plt.title('Detected Point'), plt.xticks([]), plt.yticks([]) 
    plt.suptitle(meth) 
    plt.show() 

有2个问题,这种方法:

1)不正确地检测实际的对象。 2)我想让代码告诉我哪些图片不匹配。

请找到我在下面使用的图像。

任何人都可以请帮忙吗?任何编码示例参考都可以。

谢谢!

enter image description here

enter image description here

+0

我已经更新的问题。 – Beta

+0

嗨,任何人都可以解释为什么我的问题仍然搁置。我想这是一个明确定义的问题。 – Beta

+0

我删除了我的答案,可能会让其他人更愿意回答你。祝你好运。 –

回答

0

使用机器学习的图像中检测果酱瓶里。首先使用正面和负面训练示例训练您的系统,然后使用训练好的模型预测图像是否包含果酱罐。

您可以使用CNN,SVM来达到此目的。 见链接:
http://www.pyimagesearch.com/2015/11/09/pedestrian-detection-opencv/
HOG training and detection in Python using OpenCV
http://docs.opencv.org/2.4/modules/gpu/doc/object_detection.html

+0

感谢您的评论!我会研究你的建议。 – Beta