Как найти файл / данные из заданного набора данных в проекте обработки изображений python-opencv?

Question

У меня есть набор данных изображений в проекте обработки изображений.Я хочу ввести изображение и отсканировать набор данных, чтобы распознать данное изображение.Какой модуль / библиотеку / подход (например, ML) я должен использовать для идентификации моего изображения в моем коде python-opencv?

Answer 1

Чтобы найти точно такое же изображение , вам не нужны никакие ML.Изображение - это просто массив пикселей, поэтому вы можете проверить, совпадает ли массив входного изображения с изображением в вашем наборе данных.

import glob
import cv2
import numpy as np

# Read in source image (the one you want to match to others in the dataset)
source = cv2.imread('test.jpg') 

# Make a list of all the images in the dataset (I assume they are images in a directory)
filelist = glob.glob(r'C:\Users\...\Images\*.JPG')

# Loop through the images, read them in and check if an image is equal to your source
for file in filelist:
    img = cv2.imread(file)
    if np.array_equal(source, img):
        print("%s is the same image as source" %(file))
        break