Как сгруппировать / отсортировать фотографии по схожести?

Question

[weranda]

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

Для понимания приведу простой пример.
Есть фотографии котиков:

Сортируем их и получаем отсортированных по цвету котиков:

Какие есть программы для этого?

Answer 1

[Everything_is_not_so_bad]

Вы ищете готовый коталогизатор или хотите сами создать?
Если второе, то анализируете изображения:

1. находите доминантный цвет
   Как в opencv определить доминирующий цвет?
   https://stackoverrun.com/ru/q/7910222
   
2. уровень освещенности
   https://opencv-python-tutroals.readthedocs.io/en/l...
   
3. получаете описание главного объекта продвинутым алгоритмом
   https://www.pyimagesearch.com/2017/09/11/object-de...
   

И кластеризуете по полученным характеристикам.
https://api-2d3d-cad.com/ident_by_color_texture/
https://habr.com/ru/post/321216/

Comments

[weranda]

Самому написать, боюсь, времени особо нет. Ищу готовый инструмент, а за подсказку, все равно, спасибо.

[Dimonchik]

готового нет, тем более в такой постановке

[kocherman]

Как в opencv определить доминирующий цвет?

Вы не находите глупым использование opencv для нахождения среднего цвета? Ведь алгоритм нахождения среднего арифметического значения в массиве проходят даже в школе. А использовать для этого библиотеку с нейронной сетью... такое себе...

[Владимир Куц]

kocherman, а что плохого в использовании OpenCV, если она оптимизирована под работу с изображениями, и далеко не все там завязано на нейронки.
Вручную костылить цикл на высокоуровневом языке в продакшене вместо использования биндинга к специализированной библиотеке написанной на С/C++ и оптимизированной к работе с изображениями - тоже такое себе...

[kocherman]

Владимир Куц, что именно оптимизировано в opencv, что простой цикл там работает быстрее?
Вы реально считаете, чужой код работает быстрее и стабильнее, чем тот код что напишете вы сами?

[Makssof]

kocherman, только вот нахождение доминанта не есть среднее арифметическое. С остальным, в общем-то, согласен

[kocherman]

Makssof, Ну да, алгоритм немного другой, но тоже простой как две копейки. Есть ли смысл подключать для такой задачи многомегабайтные библиотеки вместо написания 5-10 строк кода?

Answer 2

[kocherman]

Есть простые способы. Например, пакетно сжимаете все картинки до разрешения, скажем, 3х3 или 5х5 пикселей (в зависимости от количества картинок). Потом выписываете числами сначала пиксели центральные, средние и в последнюю очередь - крайние. Далее конвертируете каждый пиксель из RGB в HSL или HSV, записывая значение каждого канала в отдельное число по каждому каналу. В итоге каждой картинке будет соответствовать три длинных числа, по которым можно сортировать картинки, и они будут похожи по цветовым признакам. Картинки, где три числа одинаковые, можно считать похожими дубликатами. Если написал не достаточно толково, пишите комментарий, я могу разжевать алгоритм схематично.

UPD
Еще вспомнил про утилиту для сравнивания изображений, входящую в комплект ImageMagick. Даже в закладках сохранился на неё небольшой гайд. Но, в своё время у меня с ней не заладилось, так как работала слишком медленно а качество предоставляемой магии было не лучше скриптов, которые писал сам. Но, может, все изменилось и стоит попробовать тулзу в действии еще раз...

Answer 3

[Otrivin]

В пакете DeepFaceLab есть батник, соотирующий фото по схожести. Работает, кажется, через xnView. Можете посмотреть