Python 将单个图像与10个或更多图像进行比较，找到匹配的图像

当我运行程序时，我需要我的代码：

初始化相机

拍照

请求用户输入要存储的当前图像和要与之比较的图像的路径

检测当前拍摄照片的边缘并保存在数据库中

将当前边缘图像与数据库中的10个或更多边缘图像进行比较

输出为与当前边缘图像具有最高匹配百分比的边缘图像

基本上它就像一个物体识别程序。。。有人能帮我吗

这是我到目前为止所做的代码

from itertools import izip

import numpy as np
import cv2
from matplotlib import pyplot as plt
from PIL import Image



def take_and_save_picture(im_save):
  '''Take a picture and save it

  Args:
    im_save: filepath where the image should be stored
  '''
  camera_port = 0
  ramp_frames = 30
  cap = cv2.VideoCapture(camera_port)
  def get_image():
   retval, im = cap.read()
   return im

  for i in xrange(ramp_frames):
   temp = get_image()

  print("Taking image...")
  # Take the actual image we want to keep
  camera_capture = get_image()

  #im_save_tmp = im_save + '.jpg'
  im_save_tmp = im_save 

  # A nice feature of the imwrite method is that it will automatically choose the
  # correct format based on the file extension you provide. Convenient!
  cv2.imwrite(im_save_tmp, camera_capture)

  # You'll want to release the camera, otherwise you won't be able to create a new
  # capture object until your script exits
  # del(cap)

  img1 = cv2.imread(im_save_tmp, 0)

  edges = cv2.Canny(img1, 100, 200)
  cv2.imwrite(im_save, edges)
  cv2.waitKey(0)
  cv2.destroyAllWindows()

#im1 = "/Users/Me/gop.jpg"
#im2 = "/Users/Me/aarthi.jpg"
im1 = input('enter the path of database file')
im2 = input('enter the path where captured image is to be saved')
#im1="/Users/Me/home1.png"
#im2="/Users/Me/home.png"

def compute_edges_diff(im1, im2):
  '''Compute edges diff between to image files.

  Args:
    im1: filepath to the first image
    im2: filepath to the second image

  Returns:
    float: percentage of difference between images
  '''
#for no_file1 in range(0,10):
  #template = cv2.imread('numbers1/{no_file}.png'.format(no_file=no_file1),0)
  i1 = Image.open(im1)
  i2 = Image.open(im2)
  assert i1.mode == i2.mode, "Different kinds of images."
  assert i1.size == i2.size, "Different sizes."

  pairs = izip(i1.getdata(), i2.getdata())
  if len(i1.getbands()) == 1:
      # for gray-scale jpegs
      dif = sum(abs(p1-p2) for p1,p2 in pairs)
  else:
      dif = sum(abs(c1-c2) for p1,p2 in pairs for c1,c2 in zip(p1,p2))

  ncomponents = i1.size[0] * i1.size[1] * 3
  diff = (dif / 255.0 * 100) / ncomponents
  return diff

def main():
  #capture_img = "/Users/Me/home1.png"
  capture_img = input('enter path of the file from database')
  #img_to_compare = "/Users/Me/Documents/python programs/compare/img2.jpg"
  take_and_save_picture(capture_img)
  diff = compute_edges_diff(im1, im2)
  print "Difference (percentage):", diff
  if diff > 0.5:
   print im1
  else :
   print im2

if __name__ == '__main__':
  main()

#del(cap)

---

这段代码很好用。。但我只能比较一张图片。。。我需要将当前拍摄的图像与数据库中的所有图像进行比较…

在主功能中，创建一个列表以询问图像文件的路径，将比较结果包装在for循环中：

def get_images_to_compare():
    images_to_compare = []
    while True:
        comp_img = raw_input("Path of image to compare to: ")
        if len(comp_img) <= 1:
            # break if someone just hits enter
            break
        images_to_compare.append(comp_img)
    return images_to_compare

def main():
    #capture_img = "/Users/Me/home1.png"
    capture_img = input('enter path of the file from database')
    #img_to_compare = "/Users/Me/Documents/python programs/compare/img2.jpg"
    take_and_save_picture(capture_img)
    #### you have some odd var names here, basic gist, add a for loop
    for comp_image in get_images_to_compare():
        diff = compute_edges_diff(im1, im2)
        print "Difference (percentage):", diff
        if diff > 0.5:
            print im1
        else:
            print im2

---

但这段代码每次都会询问路径。。。我需要一个代码来比较当前的图像与一个特定文件夹中的10个图像。。。如果一个文件中有10张图片，而现在只捕获了一张图片。。。我需要系统将捕获的图像与该特定文件夹中的所有10个图像进行比较，并将差异百分比最小的图像带给我…如果是这种情况，您可以将get_images_to_compare函数替换为[os.path.join（base_path，file_path）作为os.listdir（base_path）中的文件_路径，如果file_path.endswith（'.jpg'）]。。。我对python代码不是很在行。。你能告诉我正确的语法吗。。。

def main(folder_path_to_search, files_to_compare_to, source_image_path):
    #capture_img = "/Users/Me/home1.png"
    capture_img = input('enter path of the file from database')
    #img_to_compare = "/Users/Me/Documents/python programs/compare/img2.jpg"
    take_and_save_picture(capture_img)

    images_to_compare = [ os.path.join(folder_path_to_search,file_path) for file_path in os.listdir(folder_path_to_search) if file_path.endswith(files_to_compare_to) ]

    for comp_image in get_images_to_compare():
        diff = compute_edges_diff(source_image_path, comp_image)
        print "Difference (percentage):", diff, "(", source_image_path, ":", comp_image, ")"


if __name__ == '__main__':
  folder_path_to_search = raw_input("Enter folder path to search")
  files_to_compare_to   = raw_input("enter file extention to glob ex: '.jpg'")
  source_image_path     = raw_input("enter full file path of source image")
  main(folder_path_to_search, files_to_compare_to, source_image_path)