Задача
Є нафір фотографій
В ньому
- Зображення 1
- Зображення 2
Потрібно порівняти Зображення 1 із Зображенням 2
І виставити коєфіцієнт схожесті

Всі фотографії - це кольорові зобрежння товарів на білому
В парах що порівнюються бувають ідентичні і зовсім несхожі зображення
Але, навіть фотографії ідентичних товарів можуть відрізнятись як розміром самого об'єкту так і розміром білих полів навколо товару

Тобто потрібно перед порівнянням спочатку кропнути білий фон, а потім привести порівняну пару до однакового розміру

Є база приблизно 10к пар вже порівняних вручну товарів. Порівнювали руками, відповідно, більшість ок, але є і помилки
Як я розумію, по цій базі можна навчити модель

Скрипт python має викликатись із командної строки, і отримувати в якості параметрів шляхи до фоток

Зараз порівняння фоток реалізоване таким чином

from skimage.metrics import structural_similarity
import cv2
import sys

print(sys.argv[1])
print(sys.argv[2])

image1 = cv2.imread(sys.argv[1])
image2 = cv2.imread(sys.argv[2])

image1 = cv2.cvtColor(image1, cv2.COLOR_BGR2GRAY)
image2 = cv2.cvtColor(image2, cv2.COLOR_BGR2GRAY)

similarity = structural_similarity(image1, image2)
print('result~~~', similarity)

І видає не завжди логічні результати

Ставки 4 Обговорення 1