Сходство изображений с помощью Tensorflow или PyTorch

Question

Я хочу сравнить два изображения по сходству. Поскольку моя цель - сопоставить данное изображение с огромной коллекцией изображений, я хочу запустить сравнения на графическом процессоре.

Я сталкивался с функциями tf.image.ssim и tf.image.psnr, но не могу найти и только рабочие примеры. Решения в PyTorch также приветствуются. Поскольку я не очень хорошо понимаю язык CUDA и C, я не решаюсь попробовать ядра в PyCuda.

Будет ли полезно с точки зрения обработки, если я прочитаю всю коллекцию изображений и сохраню ее как Tensorflow Records для дальнейшей обработки?

Любое руководство или решение, с благодарностью. Спасибо.

Редактировать: - Я сопоставляю изображения только одного размера. Я не хочу делать простой гистограммы. Я хочу сделать реализацию SSIM или PSNR для сходства изображений. Итак, я предполагаю, что он будет похож по цвету, содержанию и т. Д.

Answer 1

Посмотрите пример на странице документа tenorflow ( ссылка ):

im1 = tf.decode_png('path/to/im1.png')
im2 = tf.decode_png('path/to/im2.png')
print(tf.image.ssim(im1, im2, max_val=255))

Это должно работать на последней версии tenorflow. Если вы используете более старые версии, tf.image.ssim вернет тензор (print не даст вам значения), но вы можете вызвать .run () для его оценки.